Added check to abort on train mode with PerfectClassifier, UnityClass…

…ifier, and FitProbClassifier
dessn · May 14, 2021 · 69db74b · 69db74b
1 parent b51d5aa
commit 69db74b
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 = []