Ensure "Data" exists before attempting to download datasets

c-salomonsen · c-salomonsen · commit e3ba385809a6 · 2025-02-23T18:19:55.000+01:00
diff --git a/tests/test_dataloaders.py b/tests/test_dataloaders.py
@@ -25,10 +25,13 @@
     ],
 )
 def test_load_data(data_name, expected):
+    data_dir = Path("Data")
+    data_dir.mkdir(exist_ok=True)
+
     dataset, _, _ = load_data(
         data_name,
         train=False,
-        data_dir=Path("Data"),
+        data_dir=data_dir,
         transform=transforms.ToTensor(),
     )
 

Original file line number	Diff line number	Diff line change
`@@ -25,10 +25,13 @@`
`25`	`25`	`],`
`26`	`26`	`)`
`27`	`27`	`def test_load_data(data_name, expected):`
	`28`	`+ data_dir = Path("Data")`
	`29`	`+ data_dir.mkdir(exist_ok=True)`
	`30`	`+`
`28`	`31`	`dataset, _, _ = load_data(`
`29`	`32`	`data_name,`
`30`	`33`	`train=False,`
`31`		`- data_dir=Path("Data"),`
	`34`	`+ data_dir=data_dir,`
`32`	`35`	`transform=transforms.ToTensor(),`
`33`	`36`	`)`
`34`	`37`