add mixin to base class

xingyaoww · Apr 21, 2024 · 9481c7b · 9481c7b
1 parent d3c829c
commit 9481c7b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/opendevin/sandbox/docker/ssh_box.py b/opendevin/sandbox/docker/ssh_box.py
@@ -15,7 +15,6 @@
 from opendevin.sandbox.sandbox import Sandbox
 from opendevin.sandbox.process import Process
 from opendevin.sandbox.docker.process import DockerProcess
-from opendevin.sandbox.plugins.mixin import PluginMixin
 from opendevin.sandbox.plugins.jupyter import JupyterRequirement
 from opendevin.schema import ConfigType
 from opendevin.utils import find_available_tcp_port
@@ -45,7 +44,7 @@
     USER_ID = os.getuid()
 
 
-class DockerSSHBox(Sandbox, PluginMixin):
+class DockerSSHBox(Sandbox):
     instance_id: str
     container_image: str
     container_name_prefix = 'opendevin-sandbox-'

diff --git a/opendevin/sandbox/sandbox.py b/opendevin/sandbox/sandbox.py
@@ -3,9 +3,10 @@
 from typing import Tuple
 
 from opendevin.sandbox.process import Process
+from opendevin.sandbox.plugins.mixin import PluginMixin
 
 
-class Sandbox(ABC):
+class Sandbox(ABC, PluginMixin):
     background_commands: Dict[int, Process] = {}
 
     @abstractmethod