Expanded commit 7b37433

jpmml · Jan 7, 2024 · f97bf9c · f97bf9c
1 parent 7b37433
commit f97bf9c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sklearn2pmml/preprocessing/__init__.py b/sklearn2pmml/preprocessing/__init__.py
@@ -98,6 +98,7 @@ def __init__(self, dtype):
 
 	def fit(self, X, y = None):
 		if self.dtype == "category":
+			X = to_numpy(X)
 			nonmissing_mask = pandas.notnull(X)
 			categories = numpy.unique(X[nonmissing_mask])
 			self.dtype_ = CategoricalDtype(categories, ordered = False)