fix imports

MannLabs · Oct 23, 2024 · d5e9d84 · d5e9d84
1 parent 548ae85
commit d5e9d84
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/alphastats/llm/llm_functions.py b/alphastats/llm/llm_functions.py
@@ -3,8 +3,8 @@
 from typing import Dict, List
 
 import pandas as pd
-from DataSet import DataSet
 
+from alphastats.DataSet import DataSet
 from alphastats.llm.enrichment_analysis import get_enrichment_data
 from alphastats.llm.uniprot_utils import get_gene_function
 

diff --git a/tests/llm/test_llm_functions.py b/tests/llm/test_llm_functions.py
@@ -8,8 +8,8 @@
 from typing import Callable, Dict
 
 import pandas as pd
-from DataSet import DataSet
 
+from alphastats.DataSet import DataSet
 from alphastats.llm.llm_functions import (
     GENERAL_FUNCTION_MAPPING,
     get_assistant_functions,