fix: make --help return before heavy runtime imports (#293)

This commit is contained in:
Darshan
2026-04-12 16:33:30 +05:30
committed by GitHub
parent e2c74bfb3c
commit cd422bbb99
+15 -2
View File
@@ -3,6 +3,20 @@
# ruff: noqa: E402 # ruff: noqa: E402
import sys
from .config import Settings
def _is_help_invocation() -> bool:
args = sys.argv[1:]
return "-h" in args or "--help" in args
# Parse and handle CLI help before importing heavyweight ML/runtime dependencies.
if _is_help_invocation():
Settings() # ty:ignore[missing-argument]
from .progress import patch_tqdm from .progress import patch_tqdm
# This patches tqdm class definitions, which must happen # This patches tqdm class definitions, which must happen
@@ -13,7 +27,6 @@ import logging
import math import math
import os import os
import random import random
import sys
import time import time
import warnings import warnings
from dataclasses import asdict from dataclasses import asdict
@@ -45,7 +58,7 @@ from rich.table import Table
from rich.traceback import install from rich.traceback import install
from .analyzer import Analyzer from .analyzer import Analyzer
from .config import QuantizationMethod, Settings from .config import QuantizationMethod
from .evaluator import Evaluator from .evaluator import Evaluator
from .model import AbliterationParameters, Model, get_model_class from .model import AbliterationParameters, Model, get_model_class
from .system import empty_cache, get_accelerator_info from .system import empty_cache, get_accelerator_info