fix test

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · May 20, 2024 · 15db5df · 15db5df
1 parent 426190f
commit 15db5df
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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