update logger, use str instead of pickle

PrimeIntellect-ai · Dec 19, 2024 · 0fa14d0 · 0fa14d0
1 parent 79e9f17
commit 0fa14d0
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/zeroband/utils/metric_logger.py b/src/zeroband/utils/metric_logger.py
@@ -1,4 +1,3 @@
-import pickle
 from typing import Any, Protocol
 import importlib
 
@@ -45,5 +44,5 @@ def log(self, metrics: dict[str, Any]):
         self.data.append(metrics)
 
     def finish(self):
-        with open(self.project, "wb") as f:
-            pickle.dump(self.data, f)
+        with open(self.project, "w") as f:
+            f.write(str(self.data))