phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update nvidia/test_supervised_fine_tuning.py to simulate an available non-llama model

Browse source
This commit is contained in:
Matthew Farrellee 2025-07-11 13:03:31 -04:00
parent 6c16e2c0fd
commit f4af72d98e
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/unit/providers/nvidia/test_supervised_fine_tuning.py
View file

@ -7,7 +7,7 @@
import os import os
import unittest import unittest
import warnings import warnings
from unittest.mock import patch from unittest.mock import AsyncMock, patch
import pytest import pytest
@ -342,7 +342,11 @@ class TestNvidiaPostTraining(unittest.TestCase):
provider_resource_id=model_id, provider_resource_id=model_id,
model_type=model_type, model_type=model_type,
) )
result = self.run_async(self.inference_adapter.register_model(model))
# simulate a NIM where default/job-1234 is an available model
with patch.object(self.inference_adapter, "query_available_models", new_callable=AsyncMock) as mock_query:
mock_query.return_value = [model_id]
result = self.run_async(self.inference_adapter.register_model(model))
assert result == model assert result == model
assert len(self.inference_adapter.alias_to_provider_id_map) > 1 assert len(self.inference_adapter.alias_to_provider_id_map) > 1
assert self.inference_adapter.get_provider_model_id(model.provider_model_id) == model_id assert self.inference_adapter.get_provider_model_id(model.provider_model_id) == model_id