Fix bug with pathing during dataloader tests

c-salomonsen · c-salomonsen · commit 9683fe864461 · 2025-02-20T13:23:01.000+01:00
diff --git a/tests/test_wrappers.py b/tests/test_wrappers.py
@@ -50,7 +50,7 @@ def test_load_data():
     )
 
     for name in dataset_names:
-        dataset = load_data(name, train=False, data_dir=Path.cwd(), transform=trans)
+        dataset = load_data(name, train=False, data_dir=Path.cwd() / "Data", transform=trans)
 
         im, _ = dataset.__getitem__(0)
 

Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@ def test_load_data():`
`50`	`50`	`)`
`51`	`51`
`52`	`52`	`for name in dataset_names:`
`53`		`- dataset = load_data(name, train=False, data_dir=Path.cwd(), transform=trans)`
	`53`	`+ dataset = load_data(name, train=False, data_dir=Path.cwd() / "Data", transform=trans)`
`54`	`54`
`55`	`55`	`im, _ = dataset.__getitem__(0)`
`56`	`56`