Fix test

Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>
mlflow · Apr 10, 2024 · ec8ed78 · ec8ed78
1 parent 55b931c
commit ec8ed78
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/autologging/test_autologging_utils.py b/tests/autologging/test_autologging_utils.py
@@ -800,8 +800,8 @@ def test_is_autologging_integration_supported(flavor, module_version, expected_r
         ("pyspark.ml", "3.3.0.dev0", True),
         ("pyspark.ml", "3.2.1.dev0", True),
         ("pyspark.ml", "3.1.2.dev0", True),
-        ("pyspark.ml", "3.0.1.dev0", True),
-        ("pyspark.ml", "3.0.0.dev0", True),
+        ("pyspark.ml", "3.0.1.dev0", False),
+        ("pyspark.ml", "3.0.0.dev0", False),
         ("pyspark.ml", "2.4.8.dev0", False),
     ],
 )