5897356f490bc7cabc4e51c305df54e6302cc911,optuna/integration/catalyst.py,,,#,6

Before Change

import optuna


try:
    from catalyst.dl import Callback

    _available = True
except ImportError as e:
    _import_error = e
    // CatalystPruningCallback is disabled because Catalyst is not available.
    _available = False
    Callback = object


class CatalystPruningCallback(Callback):
    Catalyst callback to prune unpromising trials.

    Args:

After Change

from optuna._imports import try_import

with try_import() as _imports:
    from catalyst.dl import Callback


class CatalystPruningCallback(Callback):
    Catalyst callback to prune unpromising trials.

    Args:

Instances