From d5bd8b2bd2d71f6e27a11b38fc6b4cded0fb4286 Mon Sep 17 00:00:00 2001 From: Andrey Velichkevich Date: Fri, 15 Mar 2024 12:37:46 +0000 Subject: [PATCH] Format Python files Signed-off-by: Andrey Velichkevich --- .github/workflows/test-python.yaml | 2 +- sdk/python/kubeflow/trainer/hf_llm_training.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/workflows/test-python.yaml b/.github/workflows/test-python.yaml index ef21699476..d99be87a47 100644 --- 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 index 6e9ea02a72..d54eee9c34 100644 --- 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,