fix recopying step for forest loss driver

allenai · Jan 8, 2025 · b3d22cf · b3d22cf
1 parent bbda270
commit b3d22cf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/.github/workflows/deploy_image_on_vm.sh b/.github/workflows/deploy_image_on_vm.sh
@@ -268,8 +268,11 @@ extra_args_model_predict="$extra_args_model_predict" \
             -v $LOCAL_INDEX_CACHE_DIR:/index_cache \
             $DOCKER_IMAGE /bin/bash -c "$COMMAND" && \
         echo "Data Extraction Complete" && \
-        gsutil -m cp -r /index_cache/* $INDEX_CACHE_DIR/ && \
-        echo "Index cache copied back to $INDEX_CACHE_DIR" && \
+        if ! gsutil -m cp -r $LOCAL_INDEX_CACHE_DIR/* $INDEX_CACHE_DIR/; then
+            echo "WARNING: Failed to copy index cache back to $INDEX_CACHE_DIR" >&2
+        else
+            echo "Successfully copied index cache back to $INDEX_CACHE_DIR"
+        fi && \
         export BEAKER_TOKEN=$(curl -H "Metadata-Flavor: Google" http://metadata.google.internal/computeMetadata/v1/instance/attributes/beaker-token) && \
         export BEAKER_ADDR=$(curl -H "Metadata-Flavor: Google" http://metadata.google.internal/computeMetadata/v1/instance/attributes/beaker-addr) && \
         curl -s '\''https://beaker.org/api/v3/release/cli?os=linux&arch=amd64'\'' | sudo tar -zxv -C /usr/local/bin ./beaker && \

diff --git a/rslp/forest_loss_driver/inference/config.py b/rslp/forest_loss_driver/inference/config.py
@@ -176,7 +176,7 @@ def _get_most_recent_friday() -> datetime:
     def _default_ds_root() -> str:
         friday = PredictPipelineConfig._get_most_recent_friday()
         dated_dataset_name = f"dataset_{friday.strftime('%Y%m%d')}"
-        return f"{os.environ.get('RSLP_PREFIX', 'gs://rslearn-eai')}/datasets/forest_loss_driver/final_test_8/prediction/{dated_dataset_name}"
+        return f"{os.environ.get('RSLP_PREFIX', 'gs://rslearn-eai')}/datasets/forest_loss_driver/final_test_9/prediction/{dated_dataset_name}"
 
     model_predict_args: ModelPredictArgs
     ds_root: str = field(default_factory=_default_ds_root)