Permutation testing and learning curve

jameschapman19 · Jun 15, 2022 · fd6a588 · fd6a588
1 parent 1199721
commit fd6a588
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/cca_zoo/models/__init__.py b/cca_zoo/models/__init__.py
@@ -15,11 +15,6 @@
 from ._rcca import rCCA, CCA, PLS
 from ._tcca import TCCA, KTCCA
 
-try:
-    from ._stochastic import StochasticPowerPLS, IncrementalPLS
-except:
-    pass
-
 __all__ = [
     "GCCA",
     "KGCCA",
@@ -40,8 +35,14 @@
     "PLS",
     "TCCA",
     "KTCCA",
-    "StochasticPowerPLS",
-    "IncrementalPLS",
 ]
 
+try:
+    from ._stochastic import StochasticPowerPLS, IncrementalPLS
+
+    __all__.extend("StochasticPowerPLS",
+                   "IncrementalPLS")
+except:
+    pass
+
 classes = __all__