Add missing PrepareListFeatures to merlin.models.tf module

NVIDIA-Merlin · Jul 6, 2023 · cd4f114 · cd4f114
1 parent 7ff1c4d
commit cd4f114
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/merlin/models/tf/__init__.py b/merlin/models/tf/__init__.py
@@ -155,6 +155,7 @@
     HashedCross,
     HashedCrossAll,
     PrepareFeatures,
+    PrepareListFeatures,
     ToDense,
     ToOneHot,
     ToSparse,
@@ -226,6 +227,7 @@
     "MatrixFactorizationBlock",
     "QueryItemIdsEmbeddingsBlock",
     "PrepareFeatures",
+    "PrepareListFeatures",
     "ToSparse",
     "ToDense",
     "ToTarget",