formatting

microsoft · Sep 28, 2023 · 72406fd · 72406fd
1 parent 441ff39
commit 72406fd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/models/test_rcf.py b/tests/models/test_rcf.py
@@ -47,5 +47,7 @@ def test_seed(self) -> None:
     def test_empirical(self) -> None:
         root = os.path.join("tests", "data", "eurosat")
         ds = EuroSAT(root=root, bands=EuroSAT.rgb_bands, split="train")
-        model = RCF(in_channels=3, features=4, kernel_size=3, mode="empirical", dataset=ds)
+        model = RCF(
+            in_channels=3, features=4, kernel_size=3, mode="empirical", dataset=ds
+        )
         model(torch.randn(2, 3, 8, 8))