UTs fix

zjgemi · Aug 1, 2024 · 1ea9069 · 1ea9069
1 parent 0acbb67
commit 1ea9069
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/utils/test_dflow_query.py b/tests/utils/test_dflow_query.py
@@ -260,11 +260,11 @@ def test_sort_slice_ops(self):
 
  def test_print_keys(self):
  expected_output = [
- " 6 : init--scheduler",
- " 7 : init--id",
- " 8 : iter-000000--prep-train",
- " 9 -> 11 : iter-000000--run-train-0000 -> iter-000000--run-train-0002",
- " 12 : iter-000000--prep-run-train",
+ " 0 : init--scheduler",
+ " 1 : init--id",
+ " 2 : iter-000000--prep-train",
+ "  3 -> 5 : iter-000000--run-train-0000 -> iter-000000--run-train-0002",
+ "  6 : iter-000000--prep-run-train",
  ]
  expected_output = "\n".join(expected_output + [""])