ops(KDP): adding direct imports

piotrlaczkowski · Mar 14, 2024 · de6f215 · de6f215
1 parent 599c75a
commit de6f215
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/kdp/__init__.py b/kdp/__init__.py
@@ -0,0 +1,17 @@
+# project plugin
+from kdp.layers_factory import PreprocessorLayerFactory
+from kdp.pipeline import FeaturePreprocessor, Pipeline, ProcessingStep
+from kdp.processor import CategoryEncodingOptions, OutputModeOptions, PreprocessingModel
+from kdp.stats import DatasetStatistics, FeatureType
+
+__all__ = [
+    "ProcessingStep",
+    "Pipeline",
+    "FeaturePreprocessor",
+    "FeatureType",
+    "DatasetStatistics",
+    "PreprocessorLayerFactory",
+    "PreprocessingModel",
+    "CategoryEncodingOptions",
+    "OutputModeOptions",
+]
diff --git a/test/__init__.py b/test/__init__.py