remove register fn

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

Author: kylesayrs

examples/multimodal_vision/llama4_example.py

@@ -19,7 +19,7 @@
 # NOTE: This restructuring is specifically required for vLLM compatibility.
 # To define custom calibration logic, create a new calibration module in
 # modeling/llama4.py that inherits from `MoECalibrationModule`, and register
-# it using the `@register_moe_calibration` decorator with the appropriate
+# it using the `@MoECalibrationModule.register` decorator with the appropriate
 # module class name (e.g., "Llama4TextMoe").
 
 DATASET_ID = "neuralmagic/calibration"

src/llmcompressor/modeling/qwen3_next_moe.py

@@ -16,13 +16,10 @@
 
 import torch
 
-from llmcompressor.modeling.moe_context import (
-    MoECalibrationModule,
-    register_moe_calibration,
-)
+from llmcompressor.modeling.moe_context import MoECalibrationModule
 
 
-@register_moe_calibration("Qwen3NextSparseMoeBlock")
+@MoECalibrationModule.register("Qwen3NextSparseMoeBlock")
 class CalibrationQwen3NextSparseMoeBlock(MoECalibrationModule):
     from transformers import Qwen3NextConfig
     from transformers.models.qwen3_next.modeling_qwen3_next import (