fix: fix Python typings

lox-space · Nov 14, 2024 · 8640399 · 8640399
1 parent 833b95c
commit 8640399
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -18,7 +18,7 @@ lox-ephem = { path = "crates/lox-ephem", version = "0.1.0-alpha.1" }
 lox-io = { path = "crates/lox-io", version = "0.1.0-alpha.1" }
 lox-math = { path = "crates/lox-math", version = "0.1.0-alpha.2" }
 lox-orbits = { path = "crates/lox-orbits", version = "0.1.0-alpha.3" }
-lox-space = { path = "crates/lox-space", version = "0.1.0-alpha.4" }
+lox-space = { path = "crates/lox-space", version = "0.1.0-alpha.5" }
 lox-time = { path = "crates/lox-time", version = "0.1.0-alpha.1" }
 
 csv = "1.3.0"

diff --git a/crates/lox-space/Cargo.toml b/crates/lox-space/Cargo.toml
@@ -1,7 +1,7 @@
 [package]
 name = "lox-space"
 description = "The Lox toolbox for space mission analysis and design"
-version = "0.1.0-alpha.4"
+version = "0.1.0-alpha.5"
 rust-version.workspace = true
 edition.workspace = true
 license.workspace = true

diff --git a/crates/lox-space/lox_space.pyi b/crates/lox-space/lox_space.pyi
@@ -7,14 +7,19 @@ type Epoch = Literal["jd", "mjd", "j1950", "j2000"]
 type Unit = Literal["seconds", "days", "centuries"]
 type Vec3 = tuple[float, float, float]
 
+class ElevationMask:
+    def __new__(cls, azimuth: np.ndarray, elevation: np.ndarray): ...
+    @classmethod
+    def fixed(cls, min_elevation: float) -> Self: ...
+
 def find_events(func: Callable[[float], float], start: Time, times: list[float]): ...
 def find_windows(
     func: Callable[[float], float], start: Time, end: Time, times: list[float]
 ): ...
 def visibility(
     times: list[Time],
     gs: GroundLocation,
-    min_elevation: float,
+    mask: ElevationMask,
     sc: Trajectory,
     provider: UT1Provider,
 ): ...