fix UTs

Signed-off-by: zjgemi <liuxin_zijian@163.com>
zjgemi · Aug 1, 2024 · 33bac05 · 33bac05
1 parent 1ea9069
commit 33bac05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/utils/test_dflow_query.py b/tests/utils/test_dflow_query.py
@@ -269,7 +269,7 @@ def test_print_keys(self):
  expected_output = "\n".join(expected_output + [""])
 
  ret = print_keys_in_nice_format(
- dpgen_keys[:13],
+ dpgen_keys[:7],
  ["run-train", "run-lmp", "run-fp"],
  idx_fmt_len=8,
  )