Fixed failing test

Author: gabrielspmoreira

merlin/models/tf/blocks/interaction.py

@@ -294,7 +294,7 @@ def FMBlock(
     wide_input_block = wide_input_block or ParallelBlock(
         {
             "categorical": CategoryEncoding(cat_schema, output_mode="multi_hot", sparse=True),
-            "continuous": Filter(cont_schema, post=ToSparse()),
+            "continuous": Filter(cont_schema).connect(ToSparse()),
         },
         aggregation="concat",
     )
@@ -304,7 +304,7 @@ def FMBlock(
 
     fm_input_block = fm_input_block or InputBlockV2(
         cat_schema,
-        categorical=Embeddings(schema, dim=factors_dim),
+        categorical=Embeddings(cat_schema, dim=factors_dim),
         aggregation=None,
     )
     pairwise_interaction = SequentialBlock(

merlin/models/tf/models/ranking.py

@@ -8,7 +8,6 @@
 from merlin.models.tf.core.aggregation import ConcatFeatures
 from merlin.models.tf.core.base import Block
 from merlin.models.tf.core.combinators import ParallelBlock, TabularBlock
-from merlin.models.tf.core.tabular import Filter
 from merlin.models.tf.inputs.base import InputBlockV2
 from merlin.models.tf.inputs.embedding import EmbeddingOptions, Embeddings
 from merlin.models.tf.models.base import Model
@@ -216,7 +215,7 @@ def DeepFMModel(
 
     fm_tower = FMBlock(
         schema,
-        fm_input_block=input_block.connect(Filter(Tags.CATEGORICAL)),
+        fm_input_block=input_block,
         wide_input_block=wide_input_block,
     )