Fix rm file from list

ryanjameskennedy · Feb 27, 2024 · d1f43c9 · d1f43c9
1 parent 131de41
commit d1f43c9
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/jasentool/converge.py b/jasentool/converge.py
@@ -50,10 +50,8 @@ def compare_columns(self, tbdb_df, who_df, column_names):
 
     def rm_intermediary_files(self):
         files = os.listdir(self.download_dir)
-        files.remove('converged_who_fohm_tbdb.csv')
-        files.remove('unique_tbdb.csv')
-        files.remove('unique_who.csv')
-        files.remove('fohm.csv')
+        for filename in ['converged_who_fohm_tbdb.csv', 'unique_tbdb.csv', 'unique_who.csv', 'fohm.csv']:
+            files.remove(filename)
         for filename in files:
             filepath = os.path.join(self.download_dir, filename)
             if os.path.isfile(filepath):