From b9b379ebb5c6c323603aa6341d9d98bd4306b42e Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 14 Jun 2023 23:44:09 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 tests/integration_tests/test_preprocessing.py | 2 +-
 tests/integration_tests/test_trainer.py       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/integration_tests/test_preprocessing.py b/tests/integration_tests/test_preprocessing.py
index bf91631e1c5..6822777951e 100644
--- a/tests/integration_tests/test_preprocessing.py
+++ b/tests/integration_tests/test_preprocessing.py
@@ -174,7 +174,7 @@ def test_rlhf_reward_model_data_preprocessor():
     input_features = [
         text_feature(
             name=transcript_column,
-            encoder={"type": "auto_transformer", "pretrained_model_name_or_path": "bert-base-uncased"}
+            encoder={"type": "auto_transformer", "pretrained_model_name_or_path": "bert-base-uncased"},
         )
     ]
     output_features = [number_feature(name=id_column)]
diff --git a/tests/integration_tests/test_trainer.py b/tests/integration_tests/test_trainer.py
index d9fc0c41ee0..8b6a969ae71 100644
--- a/tests/integration_tests/test_trainer.py
+++ b/tests/integration_tests/test_trainer.py
@@ -228,7 +228,7 @@ def test_rlhf_reward_model_trainer(tmpdir):
     input_features = [
         text_feature(
             name=transcript_column,
-            encoder={"type": "auto_transformer", "pretrained_model_name_or_path": "bert-base-uncased"}
+            encoder={"type": "auto_transformer", "pretrained_model_name_or_path": "bert-base-uncased"},
         )
     ]
     output_features = [number_feature(name=id_column)]