Removed data container conversions

jpmml · Jan 1, 2024 · 0fb4700 · 0fb4700
1 parent 1c3b9bb
commit 0fb4700
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sklearn2pmml/decoration/__init__.py b/sklearn2pmml/decoration/__init__.py
@@ -314,7 +314,7 @@ def fit(self, X, y = None):
 		if isinstance(X, DataFrame):
 			for domain, column in zip(self.domains, X.columns):
 				if domain is not None:
-					domain.fit(X[column].values)
+					domain.fit(X[column])
 		else:
 			for domain, column in zip(self.domains, range(0, columns)):
 				if domain is not None:
@@ -328,7 +328,7 @@ def transform(self, X):
 		if isinstance(X, DataFrame):
 			for domain, column in zip(self.domains, X.columns):
 				if domain is not None:
-					X[column] = domain.transform(X[column].values)
+					X[column] = domain.transform(X[column])
 		else:
 			for domain, column in zip(self.domains, range(0, columns)):
 				if domain is not None: