Skip to content

Commit 52c2497

Browse files
committed
fix test_hugginface.py: change huggingface repo ID (use centralized constant HF_REPO_ID)
1 parent 2f5f771 commit 52c2497

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/fmpose3d_api/test_huggingface.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import pytest
2222

2323
from fmpose3d.inference_api.fmpose3d import FMPose3DInference
24+
from fmpose3d.utils.weights import HF_REPO_ID
2425

2526

2627
class TestDownloadModelWeights:
@@ -37,7 +38,7 @@ def test_calls_hf_hub_download_humans(self):
3738
api._download_model_weights()
3839

3940
mock_dl.assert_called_once_with(
40-
repo_id="deruyter92/fmpose_temp",
41+
repo_id=HF_REPO_ID,
4142
filename="fmpose3d_humans.pth",
4243
)
4344
assert api.model_weights_path == "/fake/cache/fmpose3d_humans.pth"
@@ -52,7 +53,7 @@ def test_calls_hf_hub_download_animals(self):
5253
api._download_model_weights()
5354

5455
mock_dl.assert_called_once_with(
55-
repo_id="deruyter92/fmpose_temp",
56+
repo_id=HF_REPO_ID,
5657
filename="fmpose3d_animals.pth",
5758
)
5859
assert api.model_weights_path == "/fake/cache/fmpose3d_animals.pth"

0 commit comments

Comments
 (0)