fix

yourbench_space/app.py

@@ -73,12 +73,11 @@ def update_process_status(uuid: str):
     return gr.update(value=True, label="Process Status: Running")
 
 def prepare_task(uuid: str, oauth_token: gr.OAuthToken | None, hf_dataset_name: str, _=None):
-    manager = MANAGERS.get(uuid)
     new_env = os.environ.copy()
     if oauth_token:
         new_env["HF_TOKEN"] = oauth_token.token
     new_env["DATASET_PREFIX"] = hf_dataset_name
-    manager.start_process(custom_env=new_env)
+    MANAGERS.start_process(uuid, custom_env=new_env)
 
 
 def update_hf_org_dropdown(oauth_token: gr.OAuthToken | None):
@@ -232,7 +231,7 @@ with gr.Blocks(theme=gr.themes.Default()) as app:
                         answers_df = gr.DataFrame()
                 
                     stages_table.change(
-                        update_dataset, inputs=[stages_table, hf_org_dropdown, hf_dataset_name, login_btn], outputs=[ingestion_df, summarization_df, single_hop, answers_df]
+                        update_dataset, inputs=[stages_table, hf_org_dropdown, hf_dataset_name], outputs=[ingestion_df, summarization_df, single_hop, answers_df]
                     )
 
             log_timer = gr.Timer(1.0, active=True)

yourbench_space/utils.py

@@ -52,7 +52,7 @@ def save_files(uuid: str, files: List[pathlib.Path]) -> str:
         else "No files were saved"
     )
 
-def update_dataset(stages, hf_org, hf_prefix, oauth_token: gr.OAuthToken):
+def update_dataset(stages: list, hf_org: str, hf_prefix: str, oauth_token: gr.OAuthToken):
     """
     Updates the dataset based on the provided stages and dataset configuration.
     """
@@ -105,9 +105,9 @@ class SubprocessManagerGroup:
         uid = SubprocessManagerGroup.grab_uuid(uid)
         del self.managers[uid]
 
-    def start_process(self, uid: Union[str, gr.State]):
+    def start_process(self, uid: Union[str, gr.State], custom_env: dict | None):
         uid = SubprocessManagerGroup.grab_uuid(uid)
-        self.managers[uid].start_process()
+        self.managers[uid].start_process(custom_env=custom_env)
 
     def stop_process(self, uid: Union[str, gr.State]):
         uid = SubprocessManagerGroup.grab_uuid(uid)