Workaround for nltk issue

alteryx · Aug 21, 2024 · 04b7adc · 04b7adc
1 parent 527de53
commit 04b7adc
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/evalml/tests/model_understanding_tests/test_partial_dependence.py b/evalml/tests/model_understanding_tests/test_partial_dependence.py
@@ -2881,11 +2881,14 @@ def test_partial_dependence_dfs_transformer_handling_with_multi_output_primitive
         index="index",
         make_index=True,
     )
-    X_fm, features = ft.dfs(
-        entityset=es,
-        target_dataframe_name="X",
-        trans_primitives=["LSA"],
-    )
+    try:
+        X_fm, features = ft.dfs(
+            entityset=es,
+            target_dataframe_name="X",
+            trans_primitives=["LSA"],
+        )
+    except ValueError:
+        pytest.xfail("NLTK bug within nlp-primitives")
 
     dfs_transformer = DFSTransformer(features=features)
     pipeline = RegressionPipeline(