|
16 | 16 |
|
17 | 17 | import numpy as np |
18 | 18 | import sys |
| 19 | +import os |
| 20 | +import warnings |
19 | 21 |
|
20 | 22 | from daal4py import _get__daal_link_version__ as dv |
21 | 23 | from sklearn import __version__ as sklearn_version |
|
25 | 27 | from distutils.version import LooseVersion as Version |
26 | 28 | import logging |
27 | 29 |
|
| 30 | +try: |
| 31 | + from pandas import DataFrame |
| 32 | + from pandas.core.dtypes.cast import find_common_type |
| 33 | + pandas_is_imported = True |
| 34 | +except (ImportError, ModuleNotFoundError): |
| 35 | + pandas_is_imported = False |
| 36 | + |
| 37 | +try: |
| 38 | + from daal4py.oneapi import is_in_sycl_ctxt as is_in_ctx |
| 39 | + ctx_imported = True |
| 40 | +except (ImportError, ModuleNotFoundError): |
| 41 | + ctx_imported = False |
| 42 | + |
28 | 43 |
|
29 | 44 | def set_idp_sklearn_verbose(): |
30 | | - import warnings |
31 | | - import os |
32 | 45 | logLevel = os.environ.get("IDP_SKLEARN_VERBOSE") |
33 | 46 | try: |
34 | 47 | if logLevel is not None: |
@@ -79,14 +92,10 @@ def parse_dtype(dt): |
79 | 92 |
|
80 | 93 |
|
81 | 94 | def getFPType(X): |
82 | | - try: |
83 | | - from pandas import DataFrame |
84 | | - from pandas.core.dtypes.cast import find_common_type |
| 95 | + if pandas_is_imported: |
85 | 96 | if isinstance(X, DataFrame): |
86 | 97 | dt = find_common_type(X.dtypes.tolist()) |
87 | 98 | return parse_dtype(dt) |
88 | | - except ImportError: |
89 | | - pass |
90 | 99 |
|
91 | 100 | dt = getattr(X, 'dtype', None) |
92 | 101 | return parse_dtype(dt) |
@@ -128,26 +137,24 @@ def get_patch_message(s): |
128 | 137 |
|
129 | 138 |
|
130 | 139 | def is_in_sycl_ctxt(): |
131 | | - try: |
132 | | - from daal4py.oneapi import is_in_sycl_ctxt as is_in_ctx |
| 140 | + if ctx_imported: |
133 | 141 | return is_in_ctx() |
134 | | - except ModuleNotFoundError: |
| 142 | + else: |
135 | 143 | return False |
136 | 144 |
|
137 | 145 |
|
138 | 146 | def is_DataFrame(X): |
139 | | - try: |
| 147 | + if pandas_is_imported: |
140 | 148 | from pandas import DataFrame |
141 | 149 | return isinstance(X, DataFrame) |
142 | | - except ImportError: |
| 150 | + else: |
143 | 151 | return False |
144 | 152 |
|
145 | 153 |
|
146 | 154 | def get_dtype(X): |
147 | | - try: |
148 | | - from pandas.core.dtypes.cast import find_common_type |
| 155 | + if pandas_is_imported: |
149 | 156 | return find_common_type(list(X.dtypes)) if is_DataFrame(X) else X.dtype |
150 | | - except ImportError: |
| 157 | + else: |
151 | 158 | return getattr(X, "dtype", None) |
152 | 159 |
|
153 | 160 |
|
|
0 commit comments