From 03707192629f1e2fd03494a4663184b96b5ca4ff Mon Sep 17 00:00:00 2001
From: Divakar Verma <137818590+divakar-amd@users.noreply.github.com>
Date: Fri, 14 Jun 2024 14:19:00 -0500
Subject: [PATCH] fix __init__ files (#52)

* fix __init__ files

* make yapf happy
---
 vllm/model_executor/__init__.py                  | 11 ++++-------
 vllm/model_executor/layers/fused_moe/__init__.py |  9 ++++-----
 2 files changed, 8 insertions(+), 12 deletions(-)
 mode change 100644 => 100755 vllm/model_executor/layers/fused_moe/__init__.py

diff --git a/vllm/model_executor/__init__.py b/vllm/model_executor/__init__.py
index 76523845063ed..fb98f4a6b46f4 100644
--- 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
old mode 100644
new mode 100755
index 2926c7d1c8a76..851ed919ae3b8
--- 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"
 ]