From 615d93c80644fb72053b0c4a1d0426ca520306ab Mon Sep 17 00:00:00 2001 From: Hao Zhang Date: Fri, 8 Aug 2025 16:21:55 +0800 Subject: [PATCH] Fix typing in tests. --- tests/attributes_test.py | 3 +-- tests/update_mask_test.py | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/attributes_test.py b/tests/attributes_test.py index 2271690..621bbf9 100644 --- a/tests/attributes_test.py +++ b/tests/attributes_test.py @@ -1,4 +1,3 @@ -import typing import pytest import torch from grassmann_tensor import GrassmannTensor @@ -16,7 +15,7 @@ ((), (), torch.randn([])), ((False, False, True), ((2, 2), (1, 3), (4, 0)), torch.randn([4, 4, 4])), ]) -def x(request: typing.Any) -> Initialization: +def x(request: pytest.FixtureRequest) -> Initialization: return request.param diff --git a/tests/update_mask_test.py b/tests/update_mask_test.py index 42123ba..7118fc0 100644 --- a/tests/update_mask_test.py +++ b/tests/update_mask_test.py @@ -1,4 +1,3 @@ -import typing import pytest import torch from grassmann_tensor import GrassmannTensor @@ -18,7 +17,7 @@ torch.randn([2, 4, 2], dtype=torch.float32), ), ]) -def x(request: typing.Any) -> Initialization: +def x(request: pytest.FixtureRequest) -> Initialization: return request.param