Update import statements in _omop.py files

theislab · Feb 15, 2024 · ee73841 · ee73841
1 parent 54d50fb
commit ee73841
Show file tree

Hide file tree

Showing 5 changed files with 595 additions and 4 deletions.
diff --git a/ehrdata/io/_omop.py b/ehrdata/io/_omop.py
@@ -6,7 +6,7 @@
 import pandas as pd
 from rich import print as rprint
 
-from ehrdata.utils.omop_utils import check_with_omop_cdm, get_column_types, get_table_catalog_dict, read_table
+from ehrdata.utils._omop_utils import check_with_omop_cdm, get_column_types, get_table_catalog_dict, read_table
 
 
 def init_omop(

diff --git a/ehrdata/pl/_omop.py b/ehrdata/pl/_omop.py
@@ -4,7 +4,7 @@
 import seaborn as sns
 
 from ehrdata.tl import get_concept_name
-from ehrdata.utils.omop_utils import get_column_types, map_concept_id, read_table
+from ehrdata.utils._omop_utils import get_column_types, map_concept_id, read_table
 
 
 # TODO allow users to pass features

diff --git a/ehrdata/pp/_omop.py b/ehrdata/pp/_omop.py
@@ -5,7 +5,7 @@
 import pandas as pd
 from rich import print as rprint
 
-from ehrdata.utils.omop_utils import get_column_types, get_feature_info, read_table
+from ehrdata.utils._omop_utils import get_column_types, get_feature_info, read_table
 
 
 def get_feature_statistics(

diff --git a/ehrdata/tl/_omop.py b/ehrdata/tl/_omop.py
@@ -4,7 +4,7 @@
 from anndata import AnnData
 from rich import print as rprint
 
-from ehrdata.utils.omop_utils import df_to_dict, get_column_types, read_table
+from ehrdata.utils._omop_utils import df_to_dict, get_column_types, read_table
 
 
 def get_concept_name(adata: Union[AnnData, dict], concept_id: Union[str, list], raise_error=False, verbose=True):