update docs

theislab · May 28, 2024 · 294f9e0 · 294f9e0
1 parent 24406b5
commit 294f9e0
Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/docs/api.md b/docs/api.md
@@ -10,9 +10,6 @@
     :toctree: generated
 
     io.init_omop
-    io.extract_features
-    io.extract_note
-    io.from_dataframe
     io.to_dataframe
 ```
 

diff --git a/ehrdata/io/_omop.py b/ehrdata/io/_omop.py
@@ -7,6 +7,7 @@
 from anndata import AnnData
 from dateutil.parser import ParserError
 from ehrapy.anndata import df_to_anndata
+from pandas import DataFrame
 from rich import print as rprint
 
 from ehrdata.utils._omop_utils import (
@@ -437,7 +438,7 @@ def to_dataframe(
     visit_occurrence_id: Optional[Union[str, list[str]]] = None,
     keep_na: Optional[bool] = False,
     convert_to_datetime: Optional[bool] = True,
-) -> pd.DataFrame:
+) -> DataFrame:
     """Convert data in .obsm of anndata object to dataframe
 
     Args: