def get_genome_keys(genome):

    """ 
    Helper function to get data table keys for a genome

    Params:
    -------
        genome: genome requested, string referring to assembly
    
    Returns:   
    --------
        keys: dict of keys """

    if genome == "mm10":
        
        keys = {'x':'x_um_abs',
                'y':'y_um_abs',
                'z':'z_um_abs',
                'pos': 'pos',
                'chr':'chr',
                'cluster':'cluster'
                }
    elif genome == "hg38":
        
        keys = {'x':'x_um',
                'y':'y_um',
                'z':'z_um',
                'pos': 'hg38_pos',
                'chr': 'hg38_chr',
                'cluster':'mle_cluster',
                'dim':['x_um', 'y_um','z_um']
                }
        
    else:
        raise ValueError("Genome not found.")
    
    return keys