fix format

Signed-off-by: helenxie-bit <helenxiehz@gmail.com>

sdk/python/test_e2e/test_e2e_train_api.py

@@ -12,23 +12,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from kubernetes import client, config
-
-from kubeflow.storage_initializer.hugging_face import (
-    HuggingFaceModelParams,
-    HuggingFaceTrainerParams,
-    HuggingFaceDatasetParams,
-)
-from kubeflow.training import TrainingClient
-from kubeflow.training import constants
-
 import logging
-
+import test.e2e.utils as utils
+
+from kubeflow.storage_initializer.hugging_face import HuggingFaceDatasetParams
+from kubeflow.storage_initializer.hugging_face import HuggingFaceModelParams
+from kubeflow.storage_initializer.hugging_face import HuggingFaceTrainerParams
+from kubeflow.training import constants
+from kubeflow.training import TrainingClient
+from kubernetes import client
+from kubernetes import config
 from peft import LoraConfig
 import transformers
 
-import test.e2e.utils as utils
-
 logging.basicConfig(format="%(message)s")
 logging.getLogger("kubeflow.training.api.training_client").setLevel(logging.DEBUG)
 

sdk/python/test_e2e/test_e2e_train_api.sh

@@ -34,4 +34,4 @@ kubectl get pvc -n kubeflow
 echo "Available CRDs"
 kubectl get crd
 
-python test_e2e_train_api.py || (kubectl get pods -n kubeflow && exit 1)
+python test_e2e_train_api.py || (kubectl get pods -n kubeflow && exit 1)