Replaced version check code

alteryx · Jul 27, 2023 · bd6ccba · bd6ccba
1 parent d387dcb
commit bd6ccba
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/evalml/pipelines/components/transformers/preprocessing/decomposer.py b/evalml/pipelines/components/transformers/preprocessing/decomposer.py
@@ -152,7 +152,7 @@ def determine_periodicity(
 
         """
         # Only need to handle nullable types on pandas < 2. Kept for backwards compatibility with pandas 1.x.
-        if pd.__version__[0] == "1":
+        if int(pd.__version__.split(".")[0]) < 2:
             X, y = cls._handle_nullable_types(cls, X, y)
 
         def _get_rel_max_from_acf(y):

diff --git a/evalml/tests/component_tests/decomposer_tests/test_decomposer.py b/evalml/tests/component_tests/decomposer_tests/test_decomposer.py
@@ -497,7 +497,7 @@ def test_decomposer_determine_periodicity(
         pytest.param(
             False,
             marks=pytest.mark.xfail(
-                condition=pd.__version__[0] == "1",
+                condition=int(pd.__version__.split(".")[0]) < 2,
                 strict=True,
                 raises=AssertionError,
                 reason="pandas 1.x does not recognize np.Nan in Float64 subtracted_floats.",