windows specific bug with temp files

Author: Andrewq11

tests/test_datamodule.py

@@ -11,6 +11,7 @@
 --------------------------------------------------------------------------------
 """
 
+import os
 import unittest as ut
 import numpy as np
 import torch
@@ -429,15 +430,17 @@ def test_splits_file(self):
         self.assertEqual(len(ds.val_ds), len(split_val))
         self.assertEqual(len(ds.test_ds), len(split_test))
 
-        # Create a TemporaryFile to save the splits, and test the datamodule
-        with tempfile.NamedTemporaryFile(suffix=".pt", dir=self.tmp_path) as temp:
+        try: 
+            # Create a TemporaryFile to save the splits, and test the datamodule
+            temp_file = tempfile.NamedTemporaryFile(suffix=".pt", dir=self.tmp_path)
+            
             # Save the splits
-            torch.save(splits, temp)
+            torch.save(splits, temp_file)
 
             # Test the datamodule
             task_kwargs = {
                 "df_path": csv_file,
-                "splits_path": temp.name,
+                "splits_path": temp_file.name,
                 "split_val": 0.0,
                 "split_test": 0.0,
             }
@@ -474,6 +477,10 @@ def test_splits_file(self):
             )
             np.testing.assert_array_equal(ds.val_ds.smiles_offsets_tensor, ds2.val_ds.smiles_offsets_tensor)
             np.testing.assert_array_equal(ds.test_ds.smiles_offsets_tensor, ds2.test_ds.smiles_offsets_tensor)
+        
+        finally:
+                temp_file.close()
+                os.unlink(temp_file.name)
 
 
 if __name__ == "__main__":