From ce3ad14207afef3300634c9b3154d4015303f462 Mon Sep 17 00:00:00 2001 From: Dillon DuPont Date: Fri, 15 Aug 2025 08:44:56 -0400 Subject: [PATCH] Changed image override to only occur if image is not set --- libs/python/computer/computer/computer.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/libs/python/computer/computer/computer.py b/libs/python/computer/computer/computer.py index 59b074c8..93c796cb 100644 --- a/libs/python/computer/computer/computer.py +++ b/libs/python/computer/computer/computer.py @@ -88,10 +88,11 @@ class Computer: self.logger = Logger("computer", verbosity) self.logger.info("Initializing Computer...") - if os_type == "macos": - image = "macos-sequoia-cua:latest" - elif os_type == "linux": - image = "trycua/cua-ubuntu:latest" + if not image: + if os_type == "macos": + image = "macos-sequoia-cua:latest" + elif os_type == "linux": + image = "trycua/cua-ubuntu:latest" image = str(image) # Store original parameters