lint

VIDA-NYU · Apr 27, 2024 · 9d3e490 · 9d3e490
1 parent aebf4e5
commit 9d3e490
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/scripts/offline_train/tabular_classification_ppo.py b/scripts/offline_train/tabular_classification_ppo.py
@@ -82,7 +82,11 @@
         y_test = y.iloc[test_indices]
 
     # Add settings
-    automl = AutoMLClassifier(time_bound=args.time_bound, output_folder=args.output_dir, checkpoints_folder=args.save_dir)
+    automl = AutoMLClassifier(
+        time_bound=args.time_bound,
+        output_folder=args.output_dir,
+        checkpoints_folder=args.save_dir,
+    )
 
     # Perform the search
     automl.fit(X_train, y_train)