pyflamestk.eam

def universal_embedding_function(rho,*args):

    pass

def bjs_embedding_function(rho,*args):

    pass

def embedding_function(rho, name, *args):

    switcher= {

        universal: universal_embedding_function,

        bjs: bjs_embedding_function

        # you can add your own embedding function here

    }

    func = switcher.get(name, lambda: "nothing")

    return func(rho,*args)