mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-26 14:58:03 +00:00
Merge branch 'main' into allow-dynamic-models-nvidia
This commit is contained in:
commit
6173d7a308
71 changed files with 3107 additions and 2381 deletions
|
|
@ -54,7 +54,8 @@ class TestNvidiaPostTraining(unittest.TestCase):
|
|||
self.mock_client.chat.completions.create = unittest.mock.AsyncMock()
|
||||
self.inference_mock_make_request = self.mock_client.chat.completions.create
|
||||
self.inference_make_request_patcher = patch(
|
||||
"llama_stack.providers.remote.inference.nvidia.nvidia.NVIDIAInferenceAdapter._get_client",
|
||||
"llama_stack.providers.remote.inference.nvidia.nvidia.NVIDIAInferenceAdapter._client",
|
||||
new_callable=unittest.mock.PropertyMock,
|
||||
return_value=self.mock_client,
|
||||
)
|
||||
self.inference_make_request_patcher.start()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue