def gan(ddgan):
    """
    Create GAN

    Args:
        ddgan (Module): The ddgan module

    Returns:
        GAN: The GAN
    """

    kwargs = {
        "nsteps": 10,
        "ndims": 5,
        "batch_size": 40,
        "batches": 5,
        "seed": 143,
        "epochs": 2,
        "n_critic": 10,
        "gen_learning_rate": 5e-4,
        "disc_learning_rate": 5e-4,
    }

    gan = ddgan.GAN(**kwargs)
    gan.setup(find_old_model=False)
    ddgan.set_seed(gan.seed)

    return gan