Fixed model lists paths.

ilya-lavrenov · May 13, 2024 · 9887065 · 9887065
1 parent 998d1e1
commit 9887065
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/text_generation/causal_lm/cpp/continuous_batching/python/tests/test_sampling.py b/text_generation/causal_lm/cpp/continuous_batching/python/tests/test_sampling.py
@@ -189,11 +189,11 @@ def test_preemption(tmp_path, scheduler_params):
 
 
 @pytest.mark.precommit
-@pytest.mark.parametrize("model_id", get_models_list("models/precommit"))
+@pytest.mark.parametrize("model_id", get_models_list(os.path.join(os.path.dirname(os.path.realpath(__file__)), "models", "precommit")))
 def test_hf_models_precommit(tmp_path, model_id):
     run_test_pipeline(tmp_path, model_id)
 
 @pytest.mark.nightly
-@pytest.mark.parametrize("model_id", get_models_list("models/nightly"))
+@pytest.mark.parametrize("model_id", get_models_list(os.path.join(os.path.dirname(os.path.realpath(__file__)), "models", "nightly")))
 def test_hf_models_nightly(tmp_path, model_id):
     run_test_pipeline(tmp_path, model_id)