Fixed test

Author: gabrielspmoreira

merlin/models/tf/models/ranking.py

@@ -14,6 +14,7 @@
 from merlin.models.tf.models.base import Model
 from merlin.models.tf.models.utils import parse_prediction_tasks
 from merlin.models.tf.prediction_tasks.base import ParallelPredictionBlock, PredictionTask
+from merlin.models.tf.transforms.features import CategoryEncoding
 from merlin.schema import Schema, Tags