fix mypy

astronomy-commons · Nov 22, 2024 · 01a971f · 01a971f
1 parent 90c8a05
commit 01a971f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/lsdb/loaders/dataframe/from_dataframe.py b/src/lsdb/loaders/dataframe/from_dataframe.py
@@ -19,7 +19,7 @@ def from_dataframe(
     drop_empty_siblings: bool = False,
     partition_size: int | None = None,
     threshold: int | None = None,
-    margin_order: int | None = -1,
+    margin_order: int = -1,
     margin_threshold: float | None = 5.0,
     should_generate_moc: bool = True,
     moc_max_order: int = 10,

diff --git a/src/lsdb/loaders/dataframe/margin_catalog_generator.py b/src/lsdb/loaders/dataframe/margin_catalog_generator.py
@@ -27,7 +27,7 @@ class MarginCatalogGenerator:
     def __init__(
         self,
         catalog: Catalog,
-        margin_order: int | None = -1,
+        margin_order: int = -1,
         margin_threshold: float = 5.0,
         use_pyarrow_types: bool = True,
         **kwargs,