REVERT ME

mlcommons · May 1, 2024 · 5b7da41 · 5b7da41
1 parent ce20325
commit 5b7da41
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/cli/medperf/commands/training/start_event.py b/cli/medperf/commands/training/start_event.py
@@ -36,9 +36,12 @@ def create_participants_list(self):
             participant_common_name = user_email
             participants_list[participant_label] = participant_common_name
         self.participants_list = participants_list
+        self.participants_list = "\n".join(
+            [f"{p},{participants_list[p]}" for p in participants_list]
+        )
 
     def submit(self):
-        dict_pretty_print(self.participants_list)
+        print(self.participants_list)
         msg = (
             f"You are about to start an event for the training experiment {self.training_exp.name}."
             " This is the list of participants (participant label, participant common name)"

diff --git a/cli/medperf/entities/event.py b/cli/medperf/entities/event.py
@@ -24,7 +24,7 @@ class TrainingEvent(Entity, MedperfSchema):
     """
 
     training_exp: int
-    participants: dict
+    participants: str
     finished: bool = False
     finished_at: Optional[datetime]
     report: Optional[dict]
@@ -87,7 +87,7 @@ def _Entity__remote_prefilter(cls, filters: dict) -> callable:
 
     def prepare_participants_list(self):
         with open(self.participants_list_path, "w") as f:
-            yaml.dump(self.participants, f)
+            f.write(self.participants)
 
     def display_dict(self):
         return {