[Core] Allow specifying custom Executor (#6557)

2024-07-19 18:25:06 -07:00
parent 2e26564259
commit 7bd82002ae
22 changed files with 310 additions and 92 deletions
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -564,6 +564,10 @@ def get_tokenizer_pool_config(tokenizer_group_type):
        return TokenizerPoolConfig(pool_size=1,
                                   pool_type="ray",
                                   extra_config={})
+    if isinstance(tokenizer_group_type, type):
+        return TokenizerPoolConfig(pool_size=1,
+                                   pool_type=tokenizer_group_type,
+                                   extra_config={})
    raise ValueError(f"Unknown tokenizer_group_type: {tokenizer_group_type}")