fix unit test xfail

Signed-off-by: Niels Bantilan <[email protected]>
flyteorg · Mar 22, 2023 · 7a225bb · 7a225bb
1 parent 258ba3d
commit 7a225bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugins/flytekit-papermill/tests/test_spark_notebook.py b/plugins/flytekit-papermill/tests/test_spark_notebook.py
@@ -20,7 +20,7 @@ def _get_nb_path(name: str, suffix: str = "", abs: bool = True, ext: str = ".ipy
 
 def test_notebook_task_simple():
 
-    if sys.version_info[:3] == (3, 11, 0):
+    if sys.version_info[:2] >= (3, 11):
         pytest.xfail(
             "pyspark serialization issues will be addressed in version 3.4.0, issue tracking at: "
             "https://issues.apache.org/jira/browse/SPARK-41125"