Fix tests

CamaraLab · Dec 16, 2024 · bf9663a · bf9663a
1 parent 72b75d2
commit bf9663a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/cajal/fused_gw_swc.py b/src/cajal/fused_gw_swc.py
@@ -121,15 +121,15 @@ def _init_fgw_pool(
     This is a private function.
     """
     global _CELLS
-    _CELLS: list[tuple[npt.NDArray[np.float64], npt.NDArray[np.float64]]] = cells  # type: ignore[name-defined]
+    _CELLS = cells  # type: ignore[name-defined]
     global _NODE_TYPES
     _NODE_TYPES: npt.NDArray[np.int32] = node_types  # type: ignore[name-defined]
     global _WORST_CASE_GW_INCREASE
     _WORST_CASE_GW_INCREASE: Optional[float] = worst_case_gw_increase  # type: ignore[name-defined]
     global _KWARGS
-    _KWARGS: dict[str, Any] = kwargs  # type: ignore[name-defined]
+    _KWARGS = kwargs  # type: ignore[name-defined]
     global _PENALTY_DICTIONARY
-    _PENALTY_DICTIONARY: dict[tuple[int, int], float] = penalty_dictionary  # type: ignore[name-defined]
+    _PENALTY_DICTIONARY = penalty_dictionary  # type: ignore[name-defined]
 
 
 def _fgw_index(p: tuple[int, int]):

diff --git a/tests/test_run_gw.py b/tests/test_run_gw.py
@@ -1,5 +1,5 @@
-from src.cajal.run_gw import compute_gw_distance_matrix
-from src.cajal.utilities import (
+from cajal.run_gw import compute_gw_distance_matrix
+from cajal.utilities import (
     read_gw_dists,
     read_gw_couplings,
     cell_iterator_csv,