testing tools

Author: SkafteNicki

mypy.ini

@@ -1,5 +1,5 @@
 [mypy]
-python_version = 3.8
+python_version = 3.9
 warn_redundant_casts = True
 warn_unused_ignores = True
 

requirements_tests.txt

@@ -2,3 +2,5 @@ pytest
 wheel
 twine==1.13.0
 flake8
+isort
+black

stochman/curves.py

@@ -13,7 +13,7 @@ def __init__(
         end: torch.Tensor,
         num_nodes: int = 5,
         requires_grad: bool = True,
-        device = None,
+        device=None,
         *args,
         **kwargs,
     ) -> None:
@@ -41,11 +41,11 @@ def _init_params(self, *args, **kwargs) -> None:
 
     @property
     def device(self):
-        """ Returns the device of the curve. """
+        """Returns the device of the curve."""
         return self.params.device
 
     def __len__(self):
-        """ Returns the batch dimension e.g. the number of curves """
+        """Returns the batch dimension e.g. the number of curves"""
         return self.begin.shape[0]
 
     def plot(self, t0: float = 0.0, t1: float = 1.0, N: int = 100, *plot_args, **plot_kwargs):

stochman/geodesic.py

@@ -94,7 +94,7 @@ def closure():
             break
         # if k % (max_iter // 10) == 0:
         #    curve.constant_speed(manifold)
-    #curve.constant_speed(manifold)
+    # curve.constant_speed(manifold)
     return max_grad < thresh
 
 

stochman/nnj.py

@@ -51,7 +51,7 @@ def __repr__(self):
         tensor_repr = tensor_repr.replace("tensor", "jacobian")
         tensor_repr += f"\n jactype={self.jactype.value if isinstance(self.jactype, Enum) else self.jactype}"
         return tensor_repr
-    
+
     def view(self, *args, **kwargs):
         out = super().view(*args, **kwargs)
         return jacobian(out, self.jactype)
@@ -102,7 +102,7 @@ def __matmul__(self, other):
 
 
 def jacobian(tensor, jactype):
-    """ Initialize a jacobian tensor by a specified jacobian type """
+    """Initialize a jacobian tensor by a specified jacobian type"""
     return Jacobian(tensor, jactype)