Fix AutoQuantize cast lint

meenchen · meenchen · commit e490b858a514 · 2026-06-01T21:01:29.000-07:00
Signed-off-by: weimingc &lt;17592131+meenchen@users.noreply.github.com&gt;
diff --git a/modelopt/torch/quantization/model_quant.py b/modelopt/torch/quantization/model_quant.py
@@ -543,7 +543,7 @@ def forward_backward_step(model, batch) -> None:
     }
     # Disable all quantizers; AutoQuantize will enable the needed ones
     set_quantizer_by_cfg(model, [{"quantizer_name": "*", "enable": False}])
-    search_constraints = cast(ConstraintsDict, constraints or {})
+    search_constraints = cast("ConstraintsDict", constraints or {})
     searcher.search(model, search_constraints, config=search_config)
 
     return model, searcher.state_dict()

Original file line number	Diff line number	Diff line change
`@@ -543,7 +543,7 @@ def forward_backward_step(model, batch) -> None:`
`543`	`543`	`}`
`544`	`544`	`# Disable all quantizers; AutoQuantize will enable the needed ones`
`545`	`545`	`set_quantizer_by_cfg(model, [{"quantizer_name": "*", "enable": False}])`
`546`		`- search_constraints = cast(ConstraintsDict, constraints or {})`
	`546`	`+ search_constraints = cast("ConstraintsDict", constraints or {})`
`547`	`547`	`searcher.search(model, search_constraints, config=search_config)`
`548`	`548`
`549`	`549`	`return model, searcher.state_dict()`