Skip to content

Commit c9d3971

Browse files
committed
Fix import path in tests for huggingface model weight downloads
1 parent 3c84663 commit c9d3971

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/fmpose3d_api/test_huggingface.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def test_calls_hf_hub_download_humans(self):
3030
"""Human model type resolves to the correct repo/filename."""
3131
api = FMPose3DInference(device="cpu")
3232

33-
with patch("fmpose3d.fmpose3d.hf_hub_download", create=True) as mock_dl:
33+
with patch("fmpose3d.inference_api.fmpose3d.hf_hub_download", create=True) as mock_dl:
3434
mock_dl.return_value = "/fake/cache/fmpose3d_humans.pth"
3535
# Patch the import inside the method
3636
with patch.dict("sys.modules", {"huggingface_hub": MagicMock(hf_hub_download=mock_dl)}):
@@ -46,7 +46,7 @@ def test_calls_hf_hub_download_animals(self):
4646
"""Animal model type resolves to the correct repo/filename."""
4747
api = FMPose3DInference.for_animals(device="cpu")
4848

49-
with patch("fmpose3d.fmpose3d.hf_hub_download", create=True) as mock_dl:
49+
with patch("fmpose3d.inference_api.fmpose3d.hf_hub_download", create=True) as mock_dl:
5050
mock_dl.return_value = "/fake/cache/fmpose3d_animals.pth"
5151
with patch.dict("sys.modules", {"huggingface_hub": MagicMock(hf_hub_download=mock_dl)}):
5252
api._download_model_weights()

0 commit comments

Comments
 (0)