Writing json results needs to create a new files

faroit · Aug 9, 2016 · d131c12 · d131c12
1 parent 9025b17
commit d131c12
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/dsdtools/evaluate.py b/dsdtools/evaluate.py
@@ -64,9 +64,7 @@ def import_json(self, filename):
             'SDR',
             'ISR',
             'SIR',
-            'SAR',
-            'sample',
-            'subset'
+            'SAR'
         ]
         new_cols = set(df.columns.tolist()).difference(set(base_cols))
         cols = base_cols + list(new_cols)
@@ -197,7 +195,7 @@ def evaluate_track(
                         track.subset,
                         track.filename
                     )
-                    with open(track_dir + '/evaluation.json', 'w') as f:
+                    with open(track_dir + '/evaluation.json', 'w+') as f:
                         f.write(json_string)
                 except (ValueError, IOError):
                     pass