From 2900ee44654806296ae84c7657700e7d0ae4bc8b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Danie=CC=88l=20de=20Kok?= <me@danieldk.eu>
Date: Mon, 8 Apr 2024 11:07:09 +0200
Subject: [PATCH] Use small ELECTRA model for testing

---
 curated_transformers/tests/models/electra/test_encoder.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/curated_transformers/tests/models/electra/test_encoder.py b/curated_transformers/tests/models/electra/test_encoder.py
index eb51b2c2..1f5a9429 100644
--- a/curated_transformers/tests/models/electra/test_encoder.py
+++ b/curated_transformers/tests/models/electra/test_encoder.py
@@ -12,11 +12,7 @@
 @pytest.mark.parametrize("with_torch_sdp", [False, True])
 @pytest.mark.parametrize(
     "model_name",
-    [
-        "jonfd/electra-small-nordic",
-        "Maltehb/aelaectra-danish-electra-small-cased",
-        "google/electra-small-discriminator",
-    ],
+    ["explosion-testing/electra-test"],
 )
 def test_encoder(model_name: str, torch_device, with_torch_sdp):
     assert_encoder_output_equals_hf(