updating failing test on windows

datamol-io · Nov 12, 2024 · a335b6f · a335b6f
1 parent 17346f8
commit a335b6f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_datamodule.py b/tests/test_datamodule.py
@@ -16,6 +16,7 @@
 import torch
 import pandas as pd
 import tempfile
+import pytest
 
 import graphium
 from graphium.utils.fs import rm, exists, get_size
@@ -380,6 +381,7 @@ def test_datamodule_multiple_data_files(self):
 
         self.assertEqual(len(ds.train_ds), 20)
 
+    @pytest.mark.usefixtures("tmp_path")
     def test_splits_file(self, tmp_path):
         # Test single CSV files
         csv_file = "tests/data/micro_ZINC_shard_1.csv"