From 8cc0bb224f559746196358476cc3402e8a332e96 Mon Sep 17 00:00:00 2001 From: Dirk Groeneveld Date: Wed, 5 Apr 2023 17:12:18 -0700 Subject: [PATCH] Some more noodling around to make click happy --- scripts/upload_artifact.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/scripts/upload_artifact.py b/scripts/upload_artifact.py index 5c65dd999..ee41f5a6b 100644 --- a/scripts/upload_artifact.py +++ b/scripts/upload_artifact.py @@ -25,6 +25,11 @@ def main( wandb_run_path: str, files_or_directories: Tuple[Path], ): + """ + Uploads artifacts to GCS. This uploads to a hardcoded bucket in GCS, because that's where we expect to keep all the artifacts for OLMo. + + WANDB_RUN_PATH: The "Weights and Biases" run path. You get this by going to the run in wandb and clicking on the "copy run path" button. We will use this as the prefix for the paths in the GCS bucket. + """ storage_client = storage.Client() bucket = storage_client.bucket("allennlp-olmo", "ai2-allennlp") prefix = wandb_run_path.strip("/")