Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-14 00:58:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update tests/fluxion/layers/test_basics.py

Browse source 
Co-authored-by: Cédric Deltheil <355031+deltheil@users.noreply.github.com>
This commit is contained in:
Benjamin Trom 2023-12-13 15:38:35 +01:00
parent 7d9ceae274
commit e2f2e33add
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/fluxion/layers/test_basics.py
View file

@ -21,8 +21,8 @@ def test_slicing_negative_indices() -> None:
def test_none_end_slicing() -> None: def test_none_end_slicing() -> None:
x = torch.randn(2, 1000, 400) x = torch.randn(2, 1000, 400)
slicing = Slicing(dim=1, start=1) slicing_layer = Slicing(dim=1, start=1)
sliced = slicing(x) sliced = slicing_layer(x)
expected = x[:, 1:, :] expected = x[:, 1:, :]
assert torch.equal(sliced, expected) assert torch.equal(sliced, expected)