revert TYPE_CHECKING to strings

mehta-lab · Dec 21, 2024 · 2db0adb · 2db0adb
1 parent c53893c
commit 2db0adb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/waveorder/models/inplane_oriented_thick_pol3d_vector.py b/waveorder/models/inplane_oriented_thick_pol3d_vector.py
@@ -2,14 +2,11 @@
 import numpy as np
 
 from torch import Tensor
-from typing import Literal, TYPE_CHECKING
+from typing import Literal
 from torch.nn.functional import avg_pool3d, interpolate
 from waveorder import optics, sampling, stokes, util
 from waveorder.visuals.napari_visuals import add_transfer_function_to_viewer
 
-if TYPE_CHECKING:
-    import napari
-
 
 def generate_test_phantom(zyx_shape: tuple[int, int, int]) -> torch.Tensor:
     # Simulate
@@ -298,7 +295,7 @@ def calculate_singular_system(sfZYX_transfer_function):
 
 
 def visualize_transfer_function(
-    viewer: napari.Viewer,
+    viewer: "napari.Viewer",
     sfZYX_transfer_function: torch.Tensor,
     zyx_scale: tuple[float, float, float],
 ) -> None:

diff --git a/waveorder/visuals/napari_visuals.py b/waveorder/visuals/napari_visuals.py
@@ -4,12 +4,9 @@
 import numpy as np
 import torch
 
-if TYPE_CHECKING:
-    import napari
-
 
 def add_transfer_function_to_viewer(
-    viewer: napari.Viewer,
+    viewer: "napari.Viewer",
     transfer_function: torch.Tensor,
     zyx_scale: tuple[float, float, float],
     layer_name: str = "Transfer Function",