wip

modal/sandbox.py

@@ -83,6 +83,7 @@ def _new(
         unencrypted_ports: Sequence[int] = [],
         proxy: Optional[_Proxy] = None,
         _experimental_scheduler_placement: Optional[SchedulerPlacement] = None,
+        enable_memory_snapshots: bool = False,
     ) -> "_Sandbox":
         """mdmd:hidden"""
 
@@ -178,6 +179,7 @@ async def _load(self: _Sandbox, resolver: Resolver, _existing_object_id: Optiona
                 open_ports=api_pb2.PortSpecs(ports=open_ports),
                 network_access=network_access,
                 proxy_id=(proxy.object_id if proxy else None),
+                enable_memory_snapshots=enable_memory_snapshots,
             )
 
             # Note - `resolver.app_id` will be `None` for app-less sandboxes
@@ -225,6 +227,8 @@ async def create(
         unencrypted_ports: Sequence[int] = [],
         # Reference to a Modal Proxy to use in front of this Sandbox.
         proxy: Optional[_Proxy] = None,
+        # Enable memory snapshots.
+        enable_memory_snapshots: bool = False,
         _experimental_scheduler_placement: Optional[
             SchedulerPlacement
         ] = None,  # Experimental controls over fine-grained scheduling (alpha).
@@ -262,6 +266,7 @@ async def create(
             unencrypted_ports=unencrypted_ports,
             proxy=proxy,
             _experimental_scheduler_placement=_experimental_scheduler_placement,
+            enable_memory_snapshots=enable_memory_snapshots,
         )
 
         app_id: Optional[str] = None
@@ -531,7 +536,7 @@ async def exec(
         resp = await retry_transient_errors(self._client.stub.ContainerExec, req)
         by_line = bufsize == 1
         return _ContainerProcess(resp.exec_id, self._client, stdout=stdout, stderr=stderr, text=text, by_line=by_line)
-    
+
     async def snapshot(self) -> str:
         req = api_pb2.SandboxSnapshotRequest(sandbox_id=self.object_id)
         resp = await retry_transient_errors(self._client.stub.SandboxSnapshot, req)

modal_proto/api.proto

@@ -2180,6 +2180,9 @@ message Sandbox {
   NetworkAccess network_access = 22;
 
   optional string proxy_id = 23;
+
+  // Enable memory snapshots.
+  bool enable_memory_snapshots = 24;
 }
 
 message SandboxCreateRequest {