Format Python files

Signed-off-by: Andrey Velichkevich <[email protected]>
andreyvelich · Mar 15, 2024 · d5bd8b2 · d5bd8b2
1 parent 519c11a
commit d5bd8b2
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/.github/workflows/test-python.yaml b/.github/workflows/test-python.yaml
@@ -18,7 +18,7 @@ jobs:
       - name: Check Python code with Black
         uses: psf/black@stable
         with:
-          version: 23.9.1
+          version: 24.2.0
           options: --check --exclude '/*kubeflow_org_v1*|__init__.py|api_client.py|configuration.py|exceptions.py|rest.py'
           src: sdk/
 

diff --git a/sdk/python/kubeflow/trainer/hf_llm_training.py b/sdk/python/kubeflow/trainer/hf_llm_training.py
@@ -95,7 +95,6 @@ def setup_peft_model(model, lora_config):
 
 
 def train_model(model, transformer_type, train_data, eval_data, tokenizer, train_args):
-
     # Setup the Trainer.
     trainer = Trainer(
         model=model,