From 0a7f9e84f6d6564c0a6553dfd3b2b4bd29249848 Mon Sep 17 00:00:00 2001 From: raspawar Date: Wed, 2 Apr 2025 10:47:59 +0000 Subject: [PATCH] fix linting --- .../providers/nvidia/test_supervised_fine_tuning.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/unit/providers/nvidia/test_supervised_fine_tuning.py b/tests/unit/providers/nvidia/test_supervised_fine_tuning.py index db83ab2de..94e189e3e 100644 --- a/tests/unit/providers/nvidia/test_supervised_fine_tuning.py +++ b/tests/unit/providers/nvidia/test_supervised_fine_tuning.py @@ -10,6 +10,12 @@ import warnings from unittest.mock import patch import pytest +from llama_stack_client.types.algorithm_config_param import LoraFinetuningConfig, QatFinetuningConfig +from llama_stack_client.types.post_training_supervised_fine_tune_params import ( + TrainingConfig, + TrainingConfigDataConfig, + TrainingConfigOptimizerConfig, +) from llama_stack.apis.models import Model, ModelType from llama_stack.providers.remote.inference.nvidia.nvidia import NVIDIAConfig, NVIDIAInferenceAdapter @@ -20,12 +26,6 @@ from llama_stack.providers.remote.post_training.nvidia.post_training import ( NvidiaPostTrainingJob, NvidiaPostTrainingJobStatusResponse, ) -from llama_stack_client.types.algorithm_config_param import LoraFinetuningConfig, QatFinetuningConfig -from llama_stack_client.types.post_training_supervised_fine_tune_params import ( - TrainingConfig, - TrainingConfigDataConfig, - TrainingConfigOptimizerConfig, -) class TestNvidiaPostTraining(unittest.TestCase):