fix warnings doc build

theislab · Oct 14, 2024 · d7b2c12 · d7b2c12
1 parent 0b1bdc2
commit d7b2c12
Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/docs/conf.py b/docs/conf.py
@@ -145,6 +145,8 @@
     ("py:class", "duckdb.duckdb.DuckDBPyConnection"),
     # Is documented as a py:attribute instead
     ("py:class", "numpy.int64"),
+    # For now not in public facing API
+    ("py:class", "awkward.highlevel.Array"),
 ]
 
 # Redirect broken parameter annotation classes

diff --git a/src/ehrdata/io/omop/omop.py b/src/ehrdata/io/omop/omop.py
@@ -285,7 +285,7 @@ def get_time_interval_table(
     concept_ids: Literal["all"] | Sequence = "all",
     aggregation_strategy: str = "first",  # what to do if multiple obs. in 1 interval. first, last, mean, median, most_frequent for categories
     # strategy="locf",
-) -> np.array:
+) -> np.ndarray:
     """Extract measurement table of an OMOP CDM Database.
 
     Parameters