Ensured compatibility with Scikit-Learn 1.0.2 and earlier

jpmml · Sep 20, 2023 · a231304 · a231304
1 parent 80fd6c0
commit a231304
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sklearn2pmml/cross_reference/__init__.py b/sklearn2pmml/cross_reference/__init__.py
@@ -1,18 +1,19 @@
 from sklearn.base import BaseEstimator, TransformerMixin
 from sklearn.pipeline import FeatureUnion
+from sklearn2pmml.preprocessing import IdentityTransformer
 
 import numpy
 
 def make_memorizer_union(memory, names):
 	return FeatureUnion([
 		("memorizer", Memorizer(memory, names)),
-		("passthrough", "passthrough"),
+		("identity", IdentityTransformer()),
 	])
 
 def make_recaller_union(memory, names):
 	return FeatureUnion([
 		("recaller", Recaller(memory, names)),
-		("passthrough", "passthrough")
+		("identity", IdentityTransformer())
 	])
 
 class _BaseMemoryManager(BaseEstimator, TransformerMixin):