from ray.air.integrations.keras import ReportCheckpointCallback

ReportCheckpointCallback.__module__ = "ray.train.tensorflow.keras"