Update tests/fluxion/layers/test_basics.py

Co-authored-by: Cédric Deltheil <[email protected]>
finegrain-ai · Dec 13, 2023 · e2f2e33 · e2f2e33
1 parent 7d9ceae
commit e2f2e33
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/fluxion/layers/test_basics.py b/tests/fluxion/layers/test_basics.py
@@ -21,8 +21,8 @@ def test_slicing_negative_indices() -> None:
 
 def test_none_end_slicing() -> None:
     x = torch.randn(2, 1000, 400)
-    slicing = Slicing(dim=1, start=1)
-    sliced = slicing(x)
+    slicing_layer = Slicing(dim=1, start=1)
+    sliced = slicing_layer(x)
     expected = x[:, 1:, :]
     assert torch.equal(sliced, expected)