def METHOD_NAME(wandb_init):
    run = wandb_init()
    wb_image = wandb.Image(np.zeros((28, 28)))
    # windows doesnt allow a backslash in media keys right now
    if platform.system() == "Windows":
        with pytest.raises(ValueError):
            run.log({r"train\image": wb_image})
    else:
        run.log({r"train\image": wb_image})
    run.finish()
    assert True