46dba208accf08362e25de425011f82d80ccaeb7,tests/test_sdk.py,TestSDK,test_experiment_keras_callback,#TestSDK#,405

Before Change

            exp.end()
        
        // Test params match what is expected
        params_messages = []
        for msg in server_sdk_messages:
            payload = msg["payload"]
            if "params" in payload:
                params_messages.append(payload)

After Change

            keras_cb = exp.callbacks.keras
            keras_cb.on_epoch_end(0, {"val_acc": 1, "val_loss": 2})
            // Sleep 1 second due to client sampling
            time.sleep(1)
            keras_cb.on_epoch_end(1, {"val_acc": 3, "val_loss": 4})
            exp.end()

        // Test metrics match what is expected

Instances