Test 2: fix quantize_to_nvfp4 import

2026-06-03 08:21:39 +00:00
parent 60b9bbd470
commit 29d6986dd4
1 changed files with 1 additions and 1 deletions
--- a/tests/unit/test_degeneration_2_mhc_falsify.py
+++ b/tests/unit/test_degeneration_2_mhc_falsify.py
@@ -45,7 +45,7 @@ def main():
    from dsv4.layers.grouped_linear import Nvfp4GroupedLinear
    from dsv4.layers.moe import Nvfp4MoE
    from dsv4.layers.shared_expert import Nvfp4SharedExpert
-    from dsv4.ops.quantize import quantize_weight_to_nvfp4
+    from dsv4.ops.quantize import quantize_weight_to_nvfp4, quantize_to_nvfp4

    NUM_GPUS = 8
    PROMPT = "The capital of France is"