fix

Author: favyen2

rslp/__init__.py

@@ -1,3 +1 @@
 """rslearn_projects: Ai2 projects built on top of rslearn."""
-
-import rslp.utils.fs  # noqa: F401 (imported but unused)

rslp/launch_beaker.py

@@ -57,13 +57,17 @@ def launch_job(
                 value="prior-satlas",  # nosec
             ),
             EnvVar(
-                name="S3_ACCESS_KEY_ID",  # nosec
+                name="WEKA_ACCESS_KEY_ID",  # nosec
                 secret="RSLEARN_WEKA_KEY",  # nosec
             ),
             EnvVar(
-                name="S3_SECRET_ACCESS_KEY",  # nosec
+                name="WEKA_SECRET_ACCESS_KEY",  # nosec
                 secret="RSLEARN_WEKA_SECRET",  # nosec
             ),
+            EnvVar(
+                name="WEKA_ENDPOINT_URL",  # nosec
+                value="https://weka-aus.beaker.org:9000",  # nosec
+            ),
             EnvVar(
                 name="RSLP_PROJECT",  # nosec
                 value=project_id,

rslp/lightning_cli.py

@@ -10,6 +10,7 @@
 from rslearn.main import RslearnLightningCLI
 from upath import UPath
 
+import rslp.utils.fs  # noqa: F401 (imported but unused)
 from rslp import launcher_lib
 
 CHECKPOINT_DIR = "gs://{rslp_bucket}/projects/{project_id}/{experiment_id}/checkpoints/"