Merge pull request #56 from Samreay/Issue22

Added check to abort on train mode with PerfectClassifier, UnityClass…
dessn · May 14, 2021 · 98e6194 · 98e6194
2 parents b51d5aa + 69db74b
commit 98e6194
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pippin/classifiers/classifier.py b/pippin/classifiers/classifier.py
@@ -162,6 +162,10 @@ def get_num_ranseed(sim_task, lcfit_task):
             else:
                 mode = Classifier.PREDICT
 
+            # Validate that train is not used on certain classifiers
+            if mode == Classifier.TRAIN:
+                assert name not in ["PerfectClassifier", "UnityClassifier", "FitProbClassifier"], f"Can not use train mode with {name}"
+
             needs_sim, needs_lc = cls.get_requirements(options)
 
             runs = []