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

test_converter: use proper exception type

Browse source 
Follow up of #102
This commit is contained in:
Cédric Deltheil 2023-10-18 12:33:40 +02:00 committed by Cédric Deltheil
parent 6ddd901767
commit 46dd710076
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_converter.py
View file

@ -3,7 +3,7 @@ import pytest
from warnings import warn from warnings import warn
import refiners.fluxion.layers as fl import refiners.fluxion.layers as fl
from refiners.fluxion.layers.chain import Distribute from refiners.fluxion.layers.chain import ChainError, Distribute
def test_converter_device_single_tensor(test_device: torch.device) -> None: def test_converter_device_single_tensor(test_device: torch.device) -> None:
@ -66,5 +66,5 @@ def test_converter_no_parent_device_or_dtype() -> None:
tensor = torch.randn(1, 1) tensor = torch.randn(1, 1)
with pytest.raises(expected_exception=ValueError): with pytest.raises(expected_exception=ChainError):
chain(tensor) chain(tensor)