From 4433c9f2be6a2dc1bdf1370914dda908b31c71b8 Mon Sep 17 00:00:00 2001 From: Albert Zeyer Date: Thu, 31 Aug 2023 15:09:02 +0200 Subject: [PATCH] small test fix --- tests/test_TFEngine.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_TFEngine.py b/tests/test_TFEngine.py index 1634c1f2fc..ebf8ae2d9f 100644 --- a/tests/test_TFEngine.py +++ b/tests/test_TFEngine.py @@ -1158,7 +1158,9 @@ def test_engine_forward_shape_deps(): engine = Engine(config=config) engine.init_network_from_config() - engine.forward_single(dataset=init_dataset(config.typed_dict["eval"]), seq_idx=0) + eval_ds = init_dataset(config.typed_dict["eval"]) + eval_ds.init_seq_order(epoch=1) + engine.forward_single(dataset=eval_ds, seq_idx=0) def test_engine_forward_EvalLayer_tf_function():