e4a49e89ad8544749ac653c6bfde2b8dcb45e47a,optuna/integration/__init__.py,,,#,9

Before Change

from optuna.types import TYPE_CHECKING


__all__ = [
    "chainer",
    "chainermn"
    "keras",
    "lightgbm",
    "tensorflow",
    "xgboost",
    "ChainerMNStudy"
    "ChainerPruningExtension",
    "KerasPruningCallback",
    "LightGBMPruningCallback",
    "TensorFlowPruningHook",
    "XGBoostPruningCallback",
]

After Change

from optuna.types import TYPE_CHECKING


_import_structure = {
    "chainer": ["ChainerPruningExtension"],
    "chainermn": ["ChainerMNStudy"],
    "keras": ["KerasPruningCallback"],
    "lightgbm": ["LightGBMPruningCallback"],
    "tensorflow": ["TensorFlowPruningHook"],
    "xgboost": ["XGBoostPruningCallback"],
}


__all__ = list(_import_structure.keys()) + sum(_import_structure.values(), [])


if sys.version_info[0] == 2 or TYPE_CHECKING:
    from optuna.integration.chainer import ChainerPruningExtension  // NOQA

Instances