def get_hidden_layers(hidden_layers):
    """Get a list of sizes for hidden layers

    Arguments:
        hidden_layers -- Can be an int if we only want one layer or a list if
        we want more layers

    Returns:
        A list constructed from one passed int or the lsit that was passed
    """

    if type(hidden_layers) is list:
        return hidden_layers
    else:
        return [hidden_layers, hidden_layers]