export pyflowsom

Author: berombau

src/flowsom/models/__init__.py

@@ -5,3 +5,5 @@
 from .consensus_cluster import ConsensusCluster  # isort:skip
 from .flowsom_estimator import FlowSOMEstimator  # isort:skip
 from .batch_flowsom_estimator import BatchFlowSOMEstimator  # isort:skip
+from .pyFlowSOM_som_estimator import PyFlowSOM_SOMEstimator  # isort:skip
+from .pyflowsom_estimator import PyFlowSOMEstimator  # isort:skip

src/flowsom/models/pyFlowSOM_som_estimator.py

@@ -1,5 +1,4 @@
 import numpy as np
-from pyFlowSOM import map_data_to_nodes, som
 from sklearn.utils.validation import check_is_fitted
 
 from . import BaseClusterEstimator
@@ -54,6 +53,8 @@ def fit(
         according to importance
         :type importance: np.array
         """
+        from pyFlowSOM import map_data_to_nodes, som
+
         alpha = self.alpha
         X = X.astype("double")
 
@@ -77,7 +78,10 @@ def fit(
 
     def predict(self, X, y=None):
         """Predict labels using the model."""
+        from pyFlowSOM import map_data_to_nodes
+
         check_is_fitted(self)
+
         X = X.astype("double")
         clusters, dists = map_data_to_nodes(self.codes, X)
         self.labels_ = clusters.astype(int) - 1