From 15db5df823c75759a8db7b6902a28787ff7c93ec Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Mon, 20 May 2024 19:37:37 -0400 Subject: [PATCH] fix test Signed-off-by: Jinzhe Zeng --- source/tests/pt/model/test_model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/source/tests/pt/model/test_model.py b/source/tests/pt/model/test_model.py index 27ef13ec86..8fdbdaf413 100644 --- a/source/tests/pt/model/test_model.py +++ b/source/tests/pt/model/test_model.py @@ -114,6 +114,7 @@ def get_intermediate_state(self, num_steps=1): dp_lr = self._get_dp_lr() dp_ds = self._get_dp_dataset() dp_ds.add_data_requirements(dp_model.input_requirement) + dp_ds.add_data_requirements(dp_loss.label_requirement) dp_model.data_stat(dp_ds) # Build graph