diff --git a/model/src/utils/get_model_name.py b/model/src/utils/get_model_name.py index cd2b532..126d32e 100644 --- a/model/src/utils/get_model_name.py +++ b/model/src/utils/get_model_name.py @@ -10,7 +10,10 @@ def get_model_name( """Get model name from model_name file.""" assert isinstance(path, Path) assert path.exists(), f'{path} does not exist' + # read file contents with open(file=path, encoding='utf-8') as fh: model_name = fh.read() + # remove newline character + model_name = model_name.strip() logging.debug('finished') return model_name