91535e910201dc403cb2b8644a75e1132bd712fe,python/ray/tune/integration/keras.py,TuneReporterCallback,__init__,#TuneReporterCallback#Any#Any#Any#,7

Before Change

        
        self.iteration = 0
        logs = logs or {}
        if freq not in ["batch", "epoch"]:
            raise ValueError("{} not supported as a frequency.".format(freq))
        self.freq = freq
        super(TuneReporterCallback, self).__init__()

    def on_batch_end(self, batch, logs=None):

After Change

                 frequency: Union[int, List[int]] = 1,
                 on: Union[str, List[str]] = "epoch_end"):

        if isinstance(frequency, list):
            if not isinstance(on, list) or len(frequency) != len(on):
                raise ValueError(
                    "If you pass a list for checkpoint frequencies, the `on` "
                    "parameter has to be a list with the same length.")

        self._frequency = frequency

        super(_TuneCheckpointCallback, self).__init__(on)

Instances