add module_test to test_padded_targets

NVIDIA-Merlin · Jul 4, 2023 · 6295d63 · 6295d63
1 parent bb6add9
commit 6295d63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/torch/transforms/test_sequences.py b/tests/unit/torch/transforms/test_sequences.py
@@ -91,7 +91,7 @@ def test_padded_targets(self, sequence_batch, sequence_schema):
         padding_op = TabularPadding(
             schema=sequence_schema, max_sequence_length=_max_sequence_length
         )
-        padded_batch = padding_op(inputs=None, batch=sequence_batch)
+        padded_batch = module_utils.module_test(padding_op, sequence_batch)
 
         assert padded_batch.targets["target_2"].shape[1] == _max_sequence_length
         assert torch.equal(padded_batch.targets["target_1"], sequence_batch.targets["target_1"])