From 1e788409ad8fee5f697b452267acaf6ce6ec6d3a Mon Sep 17 00:00:00 2001 From: helenxie-bit Date: Tue, 10 Sep 2024 20:44:12 -0700 Subject: [PATCH] fix format Signed-off-by: helenxie-bit --- .../v1beta1/kubeflow/katib/api/katib_client_test.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py b/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py index f8ded73eeff..7a833b6dcf3 100644 --- a/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py +++ b/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py @@ -583,7 +583,7 @@ def test_tune(katib_client, test_name, kwargs, expected_output): with patch.dict( "sys.modules", - { + { "kubeflow.storage_initializer": Mock(), "kubeflow.storage_initializer.hugging_face": Mock(), "kubeflow.storage_initializer.s3": Mock(), @@ -593,8 +593,8 @@ def test_tune(katib_client, test_name, kwargs, expected_output): "kubeflow.training.utils": Mock(), "kubeflow.training.constants": Mock(), "kubeflow.training.constants.constants": Mock(), - }), \ - patch( + }, + ), patch( "kubeflow.storage_initializer.hugging_face.HuggingFaceModelParams", HuggingFaceModelParams, ), patch( @@ -614,8 +614,7 @@ def test_tune(katib_client, test_name, kwargs, expected_output): ), patch( "kubeflow.katib.utils.utils.get_trial_substitutions_from_trainer", return_value={"param": "value"}, - ), \ - patch.object( + ), patch.object( katib_client, "create_experiment", return_value=Mock() ) as mock_create_experiment: try: