Fix mypy error

TorchIO-project · Oct 5, 2024 · cd44de3 · cd44de3
1 parent dd03090
commit cd44de3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -62,7 +62,7 @@ dev = [
     "pytest",
     "pytest-cov",
     "pytest-sugar",
-    "tox",
+    "tox-uv",
     "types-Deprecated",
 ]
 doc = [

diff --git a/src/torchio/transforms/augmentation/spatial/random_affine.py b/src/torchio/transforms/augmentation/spatial/random_affine.py
@@ -154,8 +154,10 @@ def get_params(
         scaling_params = torch.as_tensor(self.sample_uniform_sextet(scales))
         if isotropic:
             scaling_params.fill_(scaling_params[0])
-        rotation_params = self.sample_uniform_sextet(degrees)
-        translation_params = self.sample_uniform_sextet(translation)
+        rotation_params = torch.as_tensor(self.sample_uniform_sextet(degrees))
+        translation_params = torch.as_tensor(
+            self.sample_uniform_sextet(translation),
+        )
         return scaling_params, rotation_params, translation_params
 
     def apply_transform(self, subject: Subject) -> Subject: