use-computer 0.0.37__tar.gz → 0.0.38__tar.gz

{use_computer-0.0.37 → use_computer-0.0.38}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: use-computer
-Version: 0.0.37
+Version: 0.0.38
 Summary: Python SDK for use.computer sandboxes across macOS, iOS simulators, Windows, and Ubuntu
 Project-URL: Homepage, https://use.computer
 Project-URL: Documentation, https://api.use.computer/docs

{use_computer-0.0.37 → use_computer-0.0.38}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "use-computer"
-version = "0.0.37"
+version = "0.0.38"
 description = "Python SDK for use.computer sandboxes across macOS, iOS simulators, Windows, and Ubuntu"
 readme = "README.md"
 requires-python = ">=3.10"

{use_computer-0.0.37 → use_computer-0.0.38}/use_computer/harbor/environment/platforms/__init__.py

@@ -127,7 +127,12 @@ class WindowsAdapter(PlatformAdapter):
     is_mounted = True
 
     async def create(self, env: Any) -> AsyncSandbox:
-        return await env._client.create(type="windows", host=env._host, version=env._version)
+        return await env._client.create(
+            type="windows",
+            host=env._host,
+            version="" if env._snapshot else env._version,
+            snapshot=env._snapshot,
+        )
 
     async def setup(self, env: Any) -> None:
         await asyncio.sleep(5.0)
@@ -157,7 +162,12 @@ class UbuntuAdapter(PlatformAdapter):
     is_mounted = True
 
     async def create(self, env: Any) -> AsyncSandbox:
-        return await env._client.create(type="ubuntu", host=env._host, version=env._version)
+        return await env._client.create(
+            type="ubuntu",
+            host=env._host,
+            version="" if env._snapshot else env._version,
+            snapshot=env._snapshot,
+        )
 
     async def setup(self, env: Any) -> None:
         await asyncio.sleep(5.0)

{use_computer-0.0.37 → use_computer-0.0.38}/use_computer/harbor/environment/platforms/ubuntu.py

@@ -120,6 +120,8 @@ async def run_setup(env: Any) -> None:
         if not local.exists():
             env.logger.warning(f"setup file missing on disk: {local}")
             continue
+        parent = sh_quote(str(Path(vm_path).parent))
+        await env.ubuntu_sandbox.shell.run(f"mkdir -p {parent}", shell="bash", timeout=60)
         env.logger.info(f"uploading {local.name} -> {vm_path} ({local.stat().st_size}B)")
         await env.ubuntu_sandbox.upload(str(local), vm_path)
 

{use_computer-0.0.37 → use_computer-0.0.38}/use_computer/harbor/environment/use_computer.py

@@ -54,6 +54,7 @@ class UseComputerEnvironment(BaseEnvironment):
         host: str = "",
         platform: str = "macos",
         version: str = "",
+        snapshot: str = "",
         device_type: str = "",
         runtime: str = "",
         **kwargs,
@@ -64,6 +65,7 @@ class UseComputerEnvironment(BaseEnvironment):
         self._platform = platform
         self._platform_adapter = adapter_for(platform)
         self._version = version or os.environ.get("USE_COMPUTER_VERSION", "")
+        self._snapshot = snapshot or os.environ.get("USE_COMPUTER_SNAPSHOT", "")
         # YAML-level iOS pin (job-adhoc.yaml `environment.kwargs.device_type` /
         # `.runtime`). Used as a fallback when the task.toml doesn't pin one.
         # Lets the same dataset target different sims by editing one YAML line