formatting

ruflab · Feb 23, 2024 · 40017ac · 40017ac
1 parent f2f9602
commit 40017ac
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/test_training.py b/tests/test_training.py
@@ -69,9 +69,7 @@ def test_training():
 
     assert workspace_latent.size() == (32, 16)
 
-    reconstructed_unimodal_latents = gw.decode(
-        workspace_latent, domains={"v", "a"}
-    )
+    reconstructed_unimodal_latents = gw.decode(workspace_latent, domains={"v", "a"})
 
     assert reconstructed_unimodal_latents.keys() == {"v", "a"}
     assert reconstructed_unimodal_latents["v"].size() == (32, 128)