From 4b2ced52baf31f0a3869b34854c09470ea8a838a Mon Sep 17 00:00:00 2001
From: saidbleik <saidbleik@outlook.com>
Date: Sat, 18 Jan 2020 18:07:39 +0000
Subject: [PATCH] rem num_labels check

---
 tests/unit/test_transformers_token_classification.py | 12 +-----------
 1 file changed, 1 insertion(+), 11 deletions(-)

diff --git a/tests/unit/test_transformers_token_classification.py b/tests/unit/test_transformers_token_classification.py
index eda90c6..b4da401 100644
--- a/tests/unit/test_transformers_token_classification.py
+++ b/tests/unit/test_transformers_token_classification.py
@@ -1,18 +1,8 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 
-import pytest
-
 from utils_nlp.common.pytorch_utils import dataloader_from_dataset
-from utils_nlp.models.transformers.named_entity_recognition import (
-    TokenClassificationProcessor,
-    TokenClassifier,    
-)
-
-
-def test_token_classifier_num_labels():
-    with pytest.raises(ValueError):
-        TokenClassifier(num_labels=1)
+from utils_nlp.models.transformers.named_entity_recognition import TokenClassificationProcessor, TokenClassifier
 
 
 def test_token_classifier_fit_predict(tmp_path, ner_test_data):