def get_file_locations():
    """
        Gets the absolute paths of each data file to graph.

        Parameters:
            None

        Return:
            paths - a dict with the following structure:
            {
                env: {
                seeds: absolute_seeds_file_path
                stable_baselines: [absolute_file_paths_to_data],
                ppo_for_beginners: [absolute_file_paths_to_data]
                }
            }
    """
    # Get the absolute path of current working directory, 
    # and append graph data to it
    data_path = os.getcwd() + '/graph_data/'

    # extract environments from envs
    envs = [env for env in os.listdir(data_path) 
                        if os.path.isdir(os.path.join(data_path,env))]

    # Stores the file names.
    # Structure will be:
    # {
    #   env: {
    #            seeds: absolute_seeds_file_path
    #            stable_baselines: [absolute_file_paths_to_data],
    #            ppo_for_beginners: [absolute_file_paths_to_data]
    #        }
    # }
    paths = {}

    for env in envs:
        # Sub-dict in env as listed above
        env_data = {}
        # Extract out absolute paths of seeds.txt and data for both codes
        for directory, _, filenames in os.walk(data_path + env):
            if 'seeds.txt' in filenames:
                env_data['seeds'] = directory + '/' + filenames[0]
            elif 'stable_baselines' in directory:
                env_data['stable_baselines'] = [directory + '/' + filename 
                                                    for filename in filenames]
            elif 'ppo_for_beginners' in directory:
                env_data['ppo_for_beginners'] = [directory + '/' + filename 
                                                    for filename in filenames]

        # save the environment data into our outer paths dict
        paths[env] = env_data

    return paths
Gets the absolute paths of each data file to graph.

Parameters:
None

Return:
paths - a dict with the following structure:
{
env: {
seeds: absolute_seeds_file_path
stable_baselines: [absolute_file_paths_to_data],
ppo_for_beginners: [absolute_file_paths_to_data]
}
}
part4/make_graph.py