Removing CRF from tagging tests

cisco · Aug 2, 2022 · 8fdc245 · 8fdc245
1 parent 6b45122
commit 8fdc245
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/models/test_tagging.py b/tests/models/test_tagging.py
@@ -271,7 +271,7 @@ def test_get_boundary_counts_sequential(
 
 @pytest.mark.parametrize(
     "model_type,params",
-    [("memm", {"penalty": "l2", "C": 10000}), ("crf", {"c1": 0.01, "c2": 0.01}), ("torch-crf", {"feat_type": "dict"}),
+    [("memm", {"penalty": "l2", "C": 10000}), ("torch-crf", {"feat_type": "dict"}),
      ("torch-crf", {"feat_type": "hash"})],
 )
 def test_view_extracted_features(kwik_e_mart_nlp, model_type, params):
@@ -311,7 +311,6 @@ def test_view_extracted_features(kwik_e_mart_nlp, model_type, params):
     "query,model_type,params",
     [
         ("Main st store hours", "memm", {"penalty": "l2", "C": 10000}),
-        ("Main st store hours", "crf", {"c1": 0.01, "c2": 0.01}),
         ("Main st store hours", "torch-crf", {"feat_type": "dict"}),
         ("Main st store hours", "torch-crf", {"feat_type": "hash"})
     ],