File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change 2121import pytest
2222
2323from fmpose3d .inference_api .fmpose3d import FMPose3DInference
24+ from fmpose3d .utils .weights import HF_REPO_ID
2425
2526
2627class 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"
You can’t perform that action at this time.
0 commit comments