Replace deprecated categorical dtype detection (#197)

neuralsorcerer · meta-codesync[bot] · commit d6775d67b443 · 2025-12-04T05:53:13.000-08:00
Summary: Replaced deprecated categorical dtype detection to eliminate pandas warnings during tests. ```text DeprecationWarning: is_categorical_dtype is deprecated and will be removed in a future version. Use isinstance(dtype, pd.CategoricalDtype) instead ``` Pull Request resolved: #197 Differential Revision: D88366966 Pulled By: talgalili fbshipit-source-id: aad43b194c83931ce19acda6ed70275b3cb25de6
diff --git a/balance/util.py b/balance/util.py
@@ -203,7 +203,7 @@ def _is_categorical_dtype(series: pd.Series) -> bool:
     dtype = series.dtype
     return (
         pd_types.is_object_dtype(dtype)
-        or pd_types.is_categorical_dtype(dtype)
+        or isinstance(dtype, pd.CategoricalDtype)
         or pd_types.is_string_dtype(dtype)
     )
 

Original file line number	Diff line number	Diff line change
`@@ -203,7 +203,7 @@ def _is_categorical_dtype(series: pd.Series) -> bool:`
`203`	`203`	`dtype = series.dtype`
`204`	`204`	`return (`
`205`	`205`	`pd_types.is_object_dtype(dtype)`
`206`		`- or pd_types.is_categorical_dtype(dtype)`
	`206`	`+ or isinstance(dtype, pd.CategoricalDtype)`
`207`	`207`	`or pd_types.is_string_dtype(dtype)`
`208`	`208`	`)`
`209`	`209`