From cf3a1ccb4faf9fcbaeeef7b489cfc39a26a273a8 Mon Sep 17 00:00:00 2001
From: Magdy Saleh <magdy@predibase.com>
Date: Tue, 17 Dec 2024 16:27:33 -0500
Subject: [PATCH] fix test

---
 integration-tests/test_classifications.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/integration-tests/test_classifications.py b/integration-tests/test_classifications.py
index 4152be6b..8b274715 100644
--- a/integration-tests/test_classifications.py
+++ b/integration-tests/test_classifications.py
@@ -28,7 +28,7 @@ def test_distilbert_ner():
 def test_bert_ner():
     config = {
         "name": "bert-ner",
-        "model_id": "dslim/bert-base-NER",
+        "model_id": "magdyks/bert-base-ner",
         "docker_args": {
             "max_input_length": 512,
             "max_batch_prefill_tokens": 512,