Skip to content

Commit

Permalink
chore: neptune file download docs update
Browse files Browse the repository at this point in the history
  • Loading branch information
RuanJohn committed Feb 27, 2024
1 parent 3661566 commit b6672b8
Showing 1 changed file with 8 additions and 4 deletions.
12 changes: 8 additions & 4 deletions marl_eval/json_tools/json_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,9 +119,12 @@ def pull_neptune_data(
Args:
project_name (str): Name of the Neptune project.
tag (List): List of tags for the experiment(s) that contain the desired JSON files.
store_directory (str, optional): Directory to store the data. Default: ./downloaded_json_data.
neptune_data_key (str, optional): Key in the neptune run where the json data is stored. Default: metrics.
tag (List): List of tags for the experiment(s) that contain the
desired JSON files.
store_directory (str, optional): Directory to store the data.
Default: ./downloaded_json_data.
neptune_data_key (str, optional): Key in the neptune run where the
json data is stored. Default: metrics.
"""
# Get the run ids
project = neptune.init_project(project=project_name)
Expand Down Expand Up @@ -149,7 +152,8 @@ def pull_neptune_data(
os.remove(file_path)
except zipfile.BadZipFile:
# If the file is not zipped continue to the next file
# as it is already downloaded.
# as it is already downloaded and doesn't need to be
# unzipped.
continue
except Exception as e:
print(f"An error occurred while unzipping or storing {file_path}: {e}")
Expand Down

0 comments on commit b6672b8

Please sign in to comment.