formating

Author: victorpoughon

src/torchlensmaker/core/sag_functions.py

@@ -1,7 +1,7 @@
 import torch
 import torch.nn as nn
 
-from typing import TypeAlias, Any, List
+from typing import TypeAlias, Any
 
 Tensor: TypeAlias = torch.Tensor
 

src/torchlensmaker/core/surfaces.py

@@ -566,17 +566,16 @@ def __init__(
             assert coefficients.dtype == dtype
             assert coefficients.dim() == 1
 
-        R_tensor = to_tensor(1./R, default_dtype=dtype)
+        R_tensor = to_tensor(1.0 / R, default_dtype=dtype)
         K_tensor = to_tensor(K, default_dtype=dtype)
         coefficients_tensor = to_tensor(coefficients, default_dtype=dtype)
 
-        sag_function = SagSum([
-            Conical(R_tensor, K_tensor),
-            Aspheric(coefficients_tensor)
-        ])
+        sag_function = SagSum(
+            [Conical(R_tensor, K_tensor), Aspheric(coefficients_tensor)]
+        )
 
         super().__init__(diameter, sag_function, collision_method, dtype)
-    
+
 
 class SphereR(LocalSurface):
     """

src/torchlensmaker/viewer/tlmviewer.py

@@ -31,7 +31,6 @@
 from torchlensmaker.core.surfaces import (
     LocalSurface,
     ImplicitSurface,
-    Plane,
 )
 
 from torchlensmaker.core.transforms import (