Ruff

Author: c-salomonsen

tests/test_dataloaders.py

@@ -36,4 +36,4 @@ def test_load_data(data_name, expected):
     assert isinstance(dataset[0][0], torch.Tensor)
     assert isinstance(
         dataset[0][1], (int, torch.Tensor, np.ndarray)
-    )  # Should probably restrict this to only int or one-hot encoded tensor or array for consistency.
+    )  # Should probably restrict this to only int or one-hot encoded tensor or array for consistency.