Skip to content

Commit

Permalink
fix format
Browse files Browse the repository at this point in the history
Signed-off-by: helenxie-bit <[email protected]>
  • Loading branch information
helenxie-bit committed Sep 5, 2024
1 parent 04a7e39 commit 8c4d65a
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions test/unit/v1beta1/tune-api/test_tune_api.py
Original file line number Diff line number Diff line change
@@ -1,19 +1,19 @@
import unittest
from unittest import TestCase
from unittest.mock import Mock
from unittest.mock import patch
from unittest.mock import Mock, patch

from kubeflow.katib import KatibClient
from kubeflow.katib import models
import kubeflow.katib as katib
from kubeflow.storage_initializer.hugging_face import HuggingFaceDatasetParams
from kubeflow.storage_initializer.hugging_face import HuggingFaceModelParams
from kubeflow.storage_initializer.hugging_face import HuggingFaceTrainerParams
import transformers
from kubeflow.katib import KatibClient, models
from kubeflow.storage_initializer.hugging_face import (
HuggingFaceDatasetParams,
HuggingFaceModelParams,
HuggingFaceTrainerParams,
)
from kubeflow.training import models as training_models
from kubernetes import client
from kubernetes.client.exceptions import ApiException
from peft import LoraConfig
import transformers


class TestTuneAPI(TestCase):
Expand Down

0 comments on commit 8c4d65a

Please sign in to comment.