[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
ludwig-ai · Jun 14, 2023 · b9b379e · b9b379e
1 parent a0808cf
commit b9b379e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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)]