diff --git a/tests/conftest.py b/tests/conftest.py index 08f9845..20788b0 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -9,14 +9,12 @@ def error() -> float: @pytest.fixture def column_0(): - # return torch.tensor([3.0, 4.5, 1.0, 2.0, 7.5, 6.4, 5.5, 6.4]) - return torch.tensor([46.2, 40.4, 44.8, 48.1, 51.2, 91.9, 38.2]) + return torch.tensor([3.0, 4.5, 1.0, 2.0, 7.5, 6.4, 5.5, 6.4]) @pytest.fixture def column_1(): - # return torch.tensor([2.7, 3.3, 1.1, 2.2, 3.8, 8.2, 4.4, 3.8]) - return torch.tensor([3.5, 3.0, 3.3, 3.6, 3.8, 8.3, 3.1]) + return torch.tensor([2.7, 3.3, 1.1, 2.2, 3.8, 8.2, 4.4, 3.8]) @pytest.fixture