c7fc32d37d74dc7ea085f4885f6f17eb213d1e68,optuna/integration/tfkeras.py,,,#,4

Before Change

    from typing import Any  // NOQA
    from typing import Dict  // NOQA

try:
    from tensorflow.keras.callbacks import Callback

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


class TFKerasPruningCallback(Callback):
    tf.keras callback to prune unpromising trials.

    This callback is intend to be compatible for TensorFlow v1 and v2,

After Change

with optuna._imports.try_import() as _imports:
    from tensorflow.keras.callbacks import Callback

if not _imports.is_successful():
    Callback = object  // NOQA


class TFKerasPruningCallback(Callback):
    tf.keras callback to prune unpromising trials.

    This callback is intend to be compatible for TensorFlow v1 and v2,

Instances