File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 ()
You can’t perform that action at this time.
0 commit comments