fix test

predibase · Dec 17, 2024 · cf3a1cc · cf3a1cc
1 parent 93dbd4a
commit cf3a1cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,