Write csv dict copy

vakker · Sep 16, 2020 · a1da98e · a1da98e
1 parent 757a994
commit a1da98e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/hetsage/utils.py b/hetsage/utils.py
@@ -80,6 +80,7 @@ def __init__(self, *args, **kwargs):
         self.csv_writer = None
 
     def write_csv(self, metrics, global_step):
+        metrics = metrics.copy()
         metrics.update({'epoch': global_step})
         if self.csv_writer is None:
             self.csv_writer = csv.DictWriter(self.csv_file, fieldnames=metrics.keys())