From 27e5aa9fed044985f517a9c39947edacd313fe2f Mon Sep 17 00:00:00 2001 From: Matteo Omenetti Date: Tue, 21 Jan 2025 08:49:42 -0500 Subject: [PATCH] added revision Signed-off-by: Matteo Omenetti --- tests/test_code_formula_predictor.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/test_code_formula_predictor.py b/tests/test_code_formula_predictor.py index d0c272c..52614f4 100644 --- a/tests/test_code_formula_predictor.py +++ b/tests/test_code_formula_predictor.py @@ -37,8 +37,7 @@ def init() -> dict: } # Download models from HF - download_path = snapshot_download(repo_id="ds4sd/CodeFormula") - artifact_path = os.path.join(download_path, "") + artifact_path = snapshot_download(repo_id="ds4sd/CodeFormula", revision="v1.0.0") init["artifact_path"] = artifact_path