Skip to content

Commit fd49f9d

Browse files
committed
Keep nested pandas to type checking only
1 parent 9fc438f commit fd49f9d

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/lsdb_rubin/tract_patch_search.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,13 @@
11
import math
2+
from typing import TYPE_CHECKING
23

3-
import nested_pandas as npd
44
from lsdb.core.search.abstract_search import AbstractSearch
55
from lsdb.core.search.box_search import box_filter
66
from lsdb.types import HCCatalogTypeVar
77

8+
if TYPE_CHECKING:
9+
import nested_pandas as npd
10+
811

912
def tract_patch_search(
1013
self,
@@ -85,7 +88,7 @@ def filter_hc_catalog(self, hc_structure: HCCatalogTypeVar):
8588
# Pass the vertices to the filter_by_box method.
8689
return hc_structure.filter_by_box(ra_range, dec_range)
8790

88-
def search_points(self, frame: npd.NestedFrame, metadata) -> npd.NestedFrame:
91+
def search_points(self, frame: "npd.NestedFrame", metadata) -> "npd.NestedFrame":
8992
"""Determine the search results within a data frame.
9093
9194
Args:

0 commit comments

Comments
 (0)