Simplifying the api

gabrielspmoreira · gabrielspmoreira · commit 84fef26ba69b · 2022-09-20T11:44:51.000-03:00
diff --git a/merlin/models/tf/blocks/interaction.py b/merlin/models/tf/blocks/interaction.py
@@ -304,7 +304,7 @@ def FMBlock(
 
     fm_input_block = fm_input_block or InputBlockV2(
         cat_schema,
-        embeddings=Embeddings(schema.select_by_tag(Tags.CATEGORICAL), dim=factors_dim),
+        embeddings=Embeddings(schema, dim=factors_dim),
         aggregation=None,
     )
     pairwise_interaction = SequentialBlock(

Original file line number	Diff line number	Diff line change
`@@ -304,7 +304,7 @@ def FMBlock(`
`304`	`304`
`305`	`305`	`fm_input_block = fm_input_block or InputBlockV2(`
`306`	`306`	`cat_schema,`
`307`		`- embeddings=Embeddings(schema.select_by_tag(Tags.CATEGORICAL), dim=factors_dim),`
	`307`	`+ embeddings=Embeddings(schema, dim=factors_dim),`
`308`	`308`	`aggregation=None,`
`309`	`309`	`)`
`310`	`310`	`pairwise_interaction = SequentialBlock(`