From 33bac052293c737cb48dd7db06e8c90510a223f4 Mon Sep 17 00:00:00 2001
From: zjgemi <liuxin_zijian@163.com>
Date: Thu, 1 Aug 2024 13:29:54 +0800
Subject: [PATCH] fix UTs

Signed-off-by: zjgemi <liuxin_zijian@163.com>
---
 tests/utils/test_dflow_query.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/utils/test_dflow_query.py b/tests/utils/test_dflow_query.py
index db551089..dad71643 100644
--- 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,
         )