fix __init__ files (opendatahub-io#52)

* fix __init__ files * make yapf happy
prarit · Jun 14, 2024 · 0370719 · 0370719
1 parent 4aba555
commit 0370719
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 12 deletions.
diff --git a/vllm/model_executor/__init__.py b/vllm/model_executor/__init__.py
@@ -1,10 +1,7 @@
-from vllm.model_executor.layers.fused_moe.fused_moe import (
-    fused_moe, get_config_file_name, invoke_fused_moe_kernel,
-    moe_align_block_size)
+from vllm.model_executor.sampling_metadata import SamplingMetadata
+from vllm.model_executor.utils import set_random_seed
 
 __all__ = [
-    "fused_moe",
-    "get_config_file_name",
-    "moe_align_block_size",
-    "invoke_fused_moe_kernel",
+    "SamplingMetadata",
+    "set_random_seed",
 ]
diff --git a/vllm/model_executor/layers/fused_moe/__init__.py b/vllm/model_executor/layers/fused_moe/__init__.py
@@ -1,9 +1,8 @@
 from vllm.model_executor.layers.fused_moe.fused_moe import (
-    fused_experts, fused_moe, fused_topk, get_config_file_name)
+    fused_experts, fused_moe, fused_topk, get_config_file_name,
+    invoke_fused_moe_kernel, moe_align_block_size)
 
 __all__ = [
-    "fused_moe",
-    "fused_topk",
-    "fused_experts",
-    "get_config_file_name",
+    "fused_moe", "fused_topk", "fused_experts", "get_config_file_name",
+    "invoke_fused_moe_kernel", "moe_align_block_size"
 ]