numpify_dictionaries.py

def convert_dict_keys_to_numpy(dictionary):
    """
    Converts each element in dictionary to numpy array
    
    Parameters:
    ----------
    dictionary : dict
        Dictionary where each element is a list that must be converted to array
    
    Returns:
    -------
    dictionary : dict
    """
    for key in dictionary.keys():
        dictionary[key] = np.array(dictionary[key])
    return dictionary