Skip to content

Commit ef161e8

Browse files
committed
fix pre-commit failure
Signed-off-by: Will Guo <willg@nvidia.com>
1 parent 19dfc40 commit ef161e8

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

modelopt/onnx/quantization/autotune/workflows.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -291,7 +291,7 @@ def region_pattern_autotuning_workflow(
291291

292292
for region_idx, region in enumerate(regions):
293293
logger.info(
294-
f"Region {region_idx + 1}/{len(regions)} (ID={region.id}, level={region.get_level()})"
294+
f"Region {region_idx + 1}/{len(regions)} (ID={region.id}, level={region.level})"
295295
)
296296

297297
if node_filter_list and not _region_matches_filter(
@@ -340,7 +340,7 @@ def region_pattern_autotuning_workflow(
340340
else:
341341
logger.info(f" Tested {schemes_tested} schemes")
342342

343-
region_model_path = models_dir / f"region_{region.id}_level_{region.get_level()}.onnx"
343+
region_model_path = models_dir / f"region_{region.id}_level_{region.level}.onnx"
344344
autotuner.export_onnx(str(region_model_path), insert_qdq=True, best=True)
345345
logger.debug(f" Saved best model: {region_model_path.name}")
346346

0 commit comments

Comments
 (0)