diff --git a/model/src/utils/__init__.py b/model/src/utils/__init__.py new file mode 100644 index 0000000..ad58834 --- /dev/null +++ b/model/src/utils/__init__.py @@ -0,0 +1 @@ +from .get_model_name import get_model_name diff --git a/model/src/utils/get_model_name.py b/model/src/utils/get_model_name.py new file mode 100644 index 0000000..cd2b532 --- /dev/null +++ b/model/src/utils/get_model_name.py @@ -0,0 +1,16 @@ +"""Definition of get_model_name function.""" + +import logging +from pathlib import Path + + +def get_model_name( + path: Path, +) -> str: + """Get model name from model_name file.""" + assert isinstance(path, Path) + assert path.exists(), f'{path} does not exist' + with open(file=path, encoding='utf-8') as fh: + model_name = fh.read() + logging.debug('finished') + return model_name