From 97baf175570e76ca36e7c0c7b5bccc32a975af01 Mon Sep 17 00:00:00 2001
From: Kangyan-Zhou <zky314343421@gmail.com>
Date: Sun, 19 Apr 2026 23:18:09 -0700
Subject: [PATCH] Fix test_modelopt_export using stale ModelConfig kwargs
 (#23214)

Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
---
 test/registered/unit/model_loader/test_modelopt_export.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/test/registered/unit/model_loader/test_modelopt_export.py b/test/registered/unit/model_loader/test_modelopt_export.py
index 8c2d634de..697af4a2c 100644
--- a/test/registered/unit/model_loader/test_modelopt_export.py
+++ b/test/registered/unit/model_loader/test_modelopt_export.py
@@ -321,11 +321,10 @@ class TestModelOptExportIntegration(unittest.TestCase):
 
         model_config = ModelConfig(
             model_path="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
-            modelopt_quant="fp8",
-            modelopt_export_path=self.export_dir,
+            quantization="modelopt_fp8",
         )
 
-        load_config = LoadConfig()
+        load_config = LoadConfig(modelopt_export_path=self.export_dir)
         device_config = DeviceConfig()
 
         # Mock the quantization and export process