diff --git a/services/inference/tests/test_inference_lib.py b/services/inference/tests/test_inference_lib.py index 0774321..763a875 100644 --- a/services/inference/tests/test_inference_lib.py +++ b/services/inference/tests/test_inference_lib.py @@ -15,6 +15,7 @@ from fastapi import HTTPException from pytest import FixtureRequest from tsfminference import TSFM_CONFIG_FILE, TSFM_MODEL_DIR +from tsfminference.dirutil import resolve_model_path from tsfminference.inference import InferenceRuntime from tsfminference.inference_payloads import ( ForecastingInferenceInput, @@ -37,7 +38,7 @@ def min_context_length(model_id): - model_path: Path = TSFM_MODEL_DIR / model_id + model_path: Path = resolve_model_path(TSFM_MODEL_DIR, model_id) assert model_path.exists(), f"{model_path} does not exist!" handler, e = ForecastingServiceHandler.load(model_id=model_id, model_path=model_path) return handler.handler_config.minimum_context_length