From fd6a588b418f60081e2c534b09e27e68649886da Mon Sep 17 00:00:00 2001
From: James Chapman <james.chapman.19@ucl.ac.uk>
Date: Wed, 15 Jun 2022 17:36:09 +0100
Subject: [PATCH] Permutation testing and learning curve

---
 cca_zoo/models/__init__.py | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/cca_zoo/models/__init__.py b/cca_zoo/models/__init__.py
index fa236562..286880ba 100644
--- 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__