Clean pkl and json files before tests

KosinskiLab · Sep 5, 2024 · cfa96dd · cfa96dd
1 parent f39c330
commit cfa96dd
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/test_features_with_templates.py b/test/test_features_with_templates.py
@@ -68,6 +68,10 @@ def run_features_generation(self, file_name, chain_id, file_extension, use_mmseq
             desc_file.write(f">{file_name}_{chain_id}, {file_name}.{file_extension}, {chain_id}\n")
 
         assert Path(f"{self.TEST_DATA_DIR}/fastas/{file_name}_{chain_id}.fasta").exists()
+        # Remove .pkl, .pkl.xz, .json, .json.xz files from the features directory
+        for ext in ['*.pkl', '*.pkl.xz', '*.json', '*.json.xz']:
+            for f in self.TEST_DATA_DIR.glob(f'features/{ext}'):
+                f.unlink()
 
         cmd = [
             'create_individual_features.py',