From 94233c4dd303b7e247b4be6453bb4084a66d24bd Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Wed, 13 May 2026 21:43:47 +0000
Subject: [PATCH] Fix __init__.py: remove private imports

---
 src/nvfp4_megamoe_kernel/__init__.py | 6 ------
 1 file changed, 6 deletions(-)

diff --git a/src/nvfp4_megamoe_kernel/__init__.py b/src/nvfp4_megamoe_kernel/__init__.py
index fbaa059e..c592bdb5 100644
--- a/src/nvfp4_megamoe_kernel/__init__.py
+++ b/src/nvfp4_megamoe_kernel/__init__.py
@@ -7,9 +7,6 @@ from nvfp4_megamoe_kernel.nvfp4_mega_moe import (
     stage_activation,
 )
 from nvfp4_megamoe_kernel.weight_transform import (
-    fold_global_scale,
-    pack_ue4m3_to_uint32,
-    interleave_l1_weights,
     transform_nvfp4_weights_for_tilelang as transform_nvfp4_weights_for_mega_moe,
 )
 from nvfp4_megamoe_kernel.symm_buffer import (
@@ -23,9 +20,6 @@ __all__ = [
     "nvfp4_mega_moe_l2",
     "stage_activation",
     "transform_nvfp4_weights_for_mega_moe",
-    "fold_global_scale",
-    "pack_ue4m3_to_uint32",
-    "interleave_l1_weights",
     "SymmBuffer",
     "get_symm_buffer_for_nvfp4_mega_moe",
 ]