Add back aga run to ui command (#110)

autogluon · Nov 11, 2024 · 21c28f7 · 21c28f7
1 parent 8a75af4
commit 21c28f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/autogluon_assistant/ui/pages/task.py b/src/autogluon_assistant/ui/pages/task.py
@@ -195,7 +195,7 @@ def run_autogluon_assistant(data_dir):
         data_dir (str): The path to the data directory.
     """
     output_filename = generate_output_filename()
-    command = ["aga", data_dir]
+    command = ["aga", "run", data_dir]
     if st.session_state.preset:
         command.extend(["--presets", PRESET_MAPPING[st.session_state.preset]])
     if st.session_state.config_overrides: