no weka on augusta

allenai · Nov 15, 2024 · 9ce243e · 9ce243e
1 parent eefb045
commit 9ce243e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pdelfin/s3_utils.py b/pdelfin/s3_utils.py
@@ -180,7 +180,7 @@ def download_directory(model_choices: List[str], local_dir: str):
     weka_choices = [path for path in model_choices if path.startswith("weka://")]
 
     # This is so hacky, but if you are on beaker/pluto, don't use weka
-    if os.environ.get("BEAKER_NODE_HOSTNAME", "").lower().startswith("pluto"):
+    if os.environ.get("BEAKER_NODE_HOSTNAME", "").lower().startswith("pluto") or os.environ.get("BEAKER_NODE_HOSTNAME", "").lower().startswith("augusta") :
         weka_choices = []
 
     other_choices = [path for path in model_choices if not path.startswith("weka://")]