From 189b6deb890d0e83e31c7d22b82ba87cb72983e5 Mon Sep 17 00:00:00 2001 From: Jash Gulabrai Date: Tue, 20 May 2025 09:58:07 -0400 Subject: [PATCH] fix: unit tests --- tests/unit/providers/nvidia/test_parameters.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/providers/nvidia/test_parameters.py b/tests/unit/providers/nvidia/test_parameters.py index ea12122a0..cc33f7609 100644 --- a/tests/unit/providers/nvidia/test_parameters.py +++ b/tests/unit/providers/nvidia/test_parameters.py @@ -131,7 +131,7 @@ class TestNvidiaParameters(unittest.TestCase): def test_required_parameters_passed(self): """Test scenario 2: When required parameters are passed.""" - required_model = "meta-llama/Llama-3.1-8B-Instruct" + required_model = "meta/llama-3.2-1b-instruct@v1.0.0+L40" required_dataset_id = "required-dataset" required_job_uuid = "required-job" @@ -190,7 +190,7 @@ class TestNvidiaParameters(unittest.TestCase): self.mock_make_request.assert_called_once() call_args = self.mock_make_request.call_args - assert call_args[1]["json"]["config"] == "meta/llama-3.1-8b-instruct" + assert call_args[1]["json"]["config"] == required_model assert call_args[1]["json"]["dataset"]["name"] == required_dataset_id def test_unsupported_parameters_warning(self):