Modify tests

ludwig-ai · Jun 14, 2023 · a0808cf · a0808cf
1 parent cae42ad
commit a0808cf
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": "gpt2"}
+ 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": "gpt2"}
+ encoder={"type": "auto_transformer", "pretrained_model_name_or_path": "bert-base-uncased"}
  )
  ]
  output_features = [number_feature(name=id_column)]