qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-03-24 16:12:03 +01:00
Code Issues Projects Releases Wiki Activity

model_zoo: resolve path names coming in from caller (CLI)

Browse source 
(to make relative paths work)
This commit is contained in:
Robert Sachunsky 2026-03-05 00:45:24 +01:00
parent 41dccb216c
commit f33fd57da8
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/eynollah/model_zoo/model_zoo.py
View file

@ -33,7 +33,7 @@ class EynollahModelZoo:
basedir: str, basedir: str,
model_overrides: Optional[List[Tuple[str, str, str]]] = None, model_overrides: Optional[List[Tuple[str, str, str]]] = None,
) -> None: ) -> None:
self.model_basedir = Path(basedir) self.model_basedir = Path(basedir).resolve()
self.logger = logging.getLogger('eynollah.model_zoo') self.logger = logging.getLogger('eynollah.model_zoo')
if not self.model_basedir.exists(): if not self.model_basedir.exists():
self.logger.warning(f"Model basedir does not exist: {basedir}. Set eynollah --model-basedir to the correct directory.") self.logger.warning(f"Model basedir does not exist: {basedir}. Set eynollah --model-basedir to the correct directory.")
@ -57,7 +57,7 @@ class EynollahModelZoo:
for model_category, model_variant, model_filename in model_overrides: for model_category, model_variant, model_filename in model_overrides:
spec = self.specs.get(model_category, model_variant) spec = self.specs.get(model_category, model_variant)
self.logger.warning("Overriding filename for model spec %s to %s", spec, model_filename) self.logger.warning("Overriding filename for model spec %s to %s", spec, model_filename)
self.specs.get(model_category, model_variant).filename = model_filename self.specs.get(model_category, model_variant).filename = str(Path(model_filename).resolve())
self._overrides += model_overrides self._overrides += model_overrides
def model_path( def model_path(