Add type annotations to custom expressions

binado · Jul 25, 2024 · d8338a7 · d8338a7
1 parent a8ba699
commit d8338a7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/gladeparser/expressions.py b/gladeparser/expressions.py
@@ -1,11 +1,12 @@
+from typing import Tuple
 import polars as pl
 
 
-def filter_by_valid_redshifts(zmin: float = 0.0):
+def filter_by_valid_redshifts(zmin: float = 0.0) -> Tuple[pl.Expr, pl.Expr]:
     return (pl.col("z_cmb") > zmin, pl.col("z_cmb").is_not_null())
 
 
-def filter_by_catalog_name(catalog_name: str):
+def filter_by_catalog_name(catalog_name: str) -> Tuple[pl.Expr]:
     return (pl.col(catalog_name).is_not_null(),)