Added branching logic to ww init in nullable types util (#4369)

* added branching logic to ww init in nullable types util
alteryx · Dec 15, 2023 · 703f258 · 703f258
1 parent a0bea2f
commit 703f258
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/docs/source/release_notes.rst b/docs/source/release_notes.rst
@@ -3,6 +3,7 @@ Release Notes
 **Future Releases**
     * Enhancements
     * Fixes
+        * Fixed bug in `_downcast_nullable_y` causing woodwork initialization issues :pr:`4369`
     * Changes
     * Documentation Changes
     * Testing Changes

diff --git a/evalml/utils/nullable_type_utils.py b/evalml/utils/nullable_type_utils.py
@@ -1,4 +1,5 @@
 import pandas as pd
+import woodwork as ww
 from woodwork.logical_types import AgeNullable, BooleanNullable, IntegerNullable
 
 DOWNCAST_TYPE_DICT = {
@@ -57,7 +58,10 @@ def _downcast_nullable_y(y, handle_boolean_nullable=True, handle_integer_nullabl
         y with any incompatible nullable types downcasted to compatible equivalents.
     """
     if y.ww.schema is None:
-        y.ww.init()
+        if isinstance(y, pd.DataFrame):
+            y.ww.init()
+        else:
+            y = ww.init_series(y)
 
     incompatible_logical_types = _get_incompatible_nullable_types(
         handle_boolean_nullable,