Fix wrong keyword argument name and set max index to None

cta-observatory · Mar 30, 2021 · 04ba81a · 04ba81a
1 parent 6d7a41f
commit 04ba81a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/protopipe/mva/train_model.py b/protopipe/mva/train_model.py
@@ -74,7 +74,7 @@ def split_data(
                 data_train_sig,
                 data_test_sig,
             ) = split_train_test(
-                ds=data_sig,
+                survived_images=data_sig,
                 train_fraction=train_fraction,
                 feature_name_list=self.feature_name_list,
                 target_name=self.target_name,
@@ -88,13 +88,13 @@ def split_data(
                 data_train_bkg,
                 data_test_bkg,
             ) = split_train_test(
-                ds=data_bkg,
+                survived_images=data_bkg,
                 train_fraction=train_fraction,
                 feature_name_list=self.feature_name_list,
                 target_name=self.target_name,
             )
 
-            max_events = -1
+            max_events = None
 
             if force_same_nsig_nbkg is True:
                 if len(X_train_bkg) <= len(X_train_sig):