fix type annotation

robert-koch-institut · Feb 6, 2024 · 2d72797 · 2d72797
1 parent 9164023
commit 2d72797
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mex/mapping/connector.py b/mex/mapping/connector.py
@@ -2,9 +2,10 @@
 from pydantic import BaseModel
 
 from mex.common.models.mapping import MAPPING_MODEL_BY_EXTRACTED_CLASS_NAME
+from mex.common.types import AssetsPath
 
 
-def get_mapping_model(path: str, model_type: type[BaseModel]) -> BaseModel:
+def get_mapping_model(path: AssetsPath, model_type: type[BaseModel]) -> BaseModel:
     """Return a mapping model with default values.
 
     Args: