Merge branch 'master' into feature/enable-ipc

StoglRobotics · Sep 3, 2024 · a42eb34 · a42eb34
2 parents c8b929d + 754f3fa
commit a42eb34
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/rtwcli/rtw_cmds/rtw_cmds/workspace/create_verb.py b/rtwcli/rtw_cmds/rtw_cmds/workspace/create_verb.py
@@ -123,6 +123,7 @@ class CreateVerbArgs:
     disable_nvidia: bool = False
     docker: bool = False
     enable_ipc: bool = False
+    disable_upgrade: bool = False
 
     @property
     def ws_name(self) -> str:
@@ -392,6 +393,12 @@ def add_arguments(self, parser: argparse.ArgumentParser, cli_name: str):
             help="Enable IPC for the docker workspace.",
             default=False,
         )
+        parser.add_argument(
+            "--disable-upgrade",
+            action="store_true",
+            help="Disable execution of 'apt-get upgrade' when creating workspace.",
+            default=False,
+        )
         parser.add_argument(
             "--ws-repos-file-name",
             type=str,
@@ -608,7 +615,7 @@ def generate_intermediate_dockerfile_content(self, create_args: CreateVerbArgs)
         return textwrap.dedent(
             f"""
             FROM {create_args.base_image_name}
-            RUN apt-get update && apt-get upgrade -y
+            RUN apt-get update {"&& apt-get upgrade -y" if not create_args.disable_upgrade else ""}
             {apt_packages_cmd}
             {python_packages_cmd}
             {rtw_clone_cmd}