From 7b743ba953b094d02cf72a5a8b80a56c546d6c4f Mon Sep 17 00:00:00 2001 From: zhanqiuhu <49648934+ZhanqiuHu@users.noreply.github.com> Date: Thu, 2 Apr 2026 15:42:09 -0400 Subject: [PATCH] [CI] Fix: pass string cache_dtype in test_register_kv_caches (#38836) --- tests/v1/kv_connector/unit/test_nixl_connector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/v1/kv_connector/unit/test_nixl_connector.py b/tests/v1/kv_connector/unit/test_nixl_connector.py index 17a70a3a5..f39b78dd2 100644 --- a/tests/v1/kv_connector/unit/test_nixl_connector.py +++ b/tests/v1/kv_connector/unit/test_nixl_connector.py @@ -1613,7 +1613,7 @@ def test_register_kv_caches( ) ] ], - cache_dtype=torch.bfloat16, + cache_dtype="bfloat16", device=torch.accelerator.current_device_index(), kernel_block_sizes=[block_size], )