def METHOD_NAME(cache_algorithm: CacheAlgorithm) -> FbgemmCacheAlgorithm:
    if cache_algorithm == CacheAlgorithm.LRU:
        return FbgemmCacheAlgorithm.LRU
    elif cache_algorithm == CacheAlgorithm.LFU:
        return FbgemmCacheAlgorithm.LFU
    else:
        raise Exception(f"Invalid cache algorithm {cache_algorithm}")