Some changes to download routines.

docker/Dockerfile

@@ -36,6 +36,7 @@ RUN micromamba install -c conda-forge opencv
 # Install OmniGibson
 RUN pip install -e .
 
+RUN echo "python -m omnigibson.scripts.setup" >> /root/.bashrc
 ENTRYPOINT []
 
 CMD ["/bin/bash"]

docker/README.md

@@ -14,9 +14,9 @@ See [here](https://www.pugetsystems.com/labs/hpc/how-to-setup-nvidia-docker-and-
 2. Build the container. **From the OmniGibson root**, run: `./docker/build_docker.sh`
 
 3. Run the container
-* To get a shell inside a container with GUI: `./docker/run_docker_gui.sh`
-* To get a jupyter notebook: `./docker/run_docker_notebook.sh`
-* To get access to a shell inside a headless container `./docker/run_docker.sh`
+* To get a shell inside a container with GUI: `sudo ./docker/run_docker_gui.sh`
+* To get a jupyter notebook: `sudo ./docker/run_docker_notebook.sh`
+* To get access to a shell inside a headless container `sudo ./docker/run_docker.sh`
 
 # Development
-To push a Docker container, run: `./docker/push_docker.sh`
+To push a Docker container, run: `sudo ./docker/push_docker.sh`

omnigibson/scripts/setup.py

@@ -14,19 +14,17 @@ def main():
     print()
 
     # Only download if the dataset path doesn't exist
-    if os.path.exists(og.og_dataset_path):
-        return
-
-    print("Downloading dataset...")
-    dataset_options = {
-        "Demo": "Download the demo OmniGibson dataset",
-        "Full": "Download the full OmniGibson dataset",
-    }
-    dataset = choose_from_options(options=dataset_options, name="dataset")
-    if dataset == "Demo":
+    if not os.path.exists(og.og_dataset_path):
+        print("Downloading dataset...")
+        # dataset_options = {
+        #     "Demo": "Download the demo OmniGibson dataset",
+        #     "Full": "Download the full OmniGibson dataset",
+        # }
+        # dataset = choose_from_options(options=dataset_options, name="dataset")
+        # if dataset == "Demo":
         download_demo_data()
-    else:
-        download_og_dataset()
+        # else:
+        #     download_og_dataset()
 
     print("Downloading assets...")
     download_assets()

omnigibson/utils/asset_utils.py

@@ -283,8 +283,7 @@ def download_assets():
     Download OmniGibson assets
     """
     if os.path.exists(og.assets_path):
-        print("Assets path already exists. Please either remove or change the asset path location from "
-              "omnigibson/global_config.yaml")
+        print("Assets already downloaded.")
     else:
         tmp_file = os.path.join(tempfile.gettempdir(), "og_assets.tar.gz")
         os.makedirs(os.path.dirname(og.assets_path), exist_ok=True)