teradataml 20.0.0.2__py3-none-any.whl → 20.0.0.3__py3-none-any.whl

teradataml/opensource/{sklearn/_wrapper_utils.py → _wrapper_utils.py}

@@ -23,10 +23,9 @@ from teradataml.common.messagecodes import MessageCodes
 from teradataml.common.messages import Messages
 from teradataml.common.aed_utils import AedUtils
 from teradataml.dataframe.dataframe_utils import DataFrameUtils
-from teradataml.dataframe.dataframe import DataFrame, in_schema
+from teradataml.dataframe.dataframe import DataFrame
 from teradataml.common.utils import UtilFuncs
 from teradataml.utils.validators import _Validators
-from teradataml.dataframe.sql_interfaces import ColumnExpression
 
 aed_utils = AedUtils()
 df_utils = DataFrameUtils()

teradataml/opensource/{sklearn/constants.py → constants.py}

@@ -16,21 +16,24 @@
 from enum import Enum
 from teradataml import VARCHAR, BLOB
 from dataclasses import dataclass, field
-from typing import Any, Dict, Tuple, Optional, List
+from typing import Any, Dict, Tuple, Optional
 
 
-_MODULES = ["sklearn.calibration", "sklearn.cluster", "sklearn.compose", "sklearn.covariance",
-            "sklearn.decomposition", "sklearn.discriminant_analysis",
-            "sklearn.dummy", "sklearn.ensemble", "sklearn.feature_extraction", "sklearn.feature_selection",
-            "sklearn.gaussian_process", "sklearn.impute", "sklearn.isotonic", "sklearn.kernel_approximation",
-            "sklearn.kernel_ridge", "sklearn.linear_model", "sklearn.manifold", "sklearn.mixture",
-            "sklearn.model_selection", "sklearn.multiclass", "sklearn.multioutput", "sklearn.naive_bayes",
-            "sklearn.neighbors", "sklearn.neural_network", "sklearn.preprocessing", "sklearn.random_projection",
-            "sklearn.semi_supervised", "sklearn.svm", "sklearn.tree", "sklearn.pipeline", "sklearn.cross_decomposition",
-            "sklearn.gaussian_process.kernels", "sklearn.metrics"]
+_SKL_MODULES = ["sklearn.calibration", "sklearn.cluster", "sklearn.compose", "sklearn.covariance",
+                "sklearn.decomposition", "sklearn.discriminant_analysis",
+                "sklearn.dummy", "sklearn.ensemble", "sklearn.feature_extraction", "sklearn.feature_selection",
+                "sklearn.gaussian_process", "sklearn.impute", "sklearn.isotonic", "sklearn.kernel_approximation",
+                "sklearn.kernel_ridge", "sklearn.linear_model", "sklearn.manifold", "sklearn.mixture",
+                "sklearn.model_selection", "sklearn.multiclass", "sklearn.multioutput", "sklearn.naive_bayes",
+                "sklearn.neighbors", "sklearn.neural_network", "sklearn.preprocessing", "sklearn.random_projection",
+                "sklearn.semi_supervised", "sklearn.svm", "sklearn.tree", "sklearn.pipeline", "sklearn.cross_decomposition",
+                "sklearn.gaussian_process.kernels", "sklearn.metrics"]
+_LIGHTGBM_MODULES = ["lightgbm.basic", "lightgbm.callback", "lightgbm.compat", "lightgbm.engine", "lightgbm.sklearn"]
+# "lightgbm.cv", "lightgbm.dask",
 
 class OpenSourcePackage(Enum):
     SKLEARN = "sklearn"
+    LIGHTGBM = "lightgbm"
 
     @classmethod
     def values(cls):

teradataml/opensource/sklearn/__init__.py

@@ -1 +0,0 @@
-from ._class import td_sklearn