isolate 0.12.0__py3-none-any.whl → 0.12.1__py3-none-any.whl

isolate/backends/conda.py

@@ -42,6 +42,7 @@ class CondaEnvironment(BaseEnvironment[Path]):
 
     environment_definition: Dict[str, Any] = field(default_factory=dict)
     python_version: Optional[str] = None
+    tags: List[str] = field(default_factory=list)
 
     @classmethod
     def from_config(
@@ -106,7 +107,11 @@ class CondaEnvironment(BaseEnvironment[Path]):
 
     @property
     def key(self) -> str:
-        return sha256_digest_of(repr(self.environment_definition))
+        return sha256_digest_of(
+            repr(self.environment_definition),
+            self.python_version,
+            *sorted(self.tags),
+        )
 
     def create(self, *, force: bool = False) -> Path:
         env_path = self.settings.cache_dir_for(self)

isolate/backends/virtualenv.py

@@ -29,6 +29,7 @@ class VirtualPythonEnvironment(BaseEnvironment[Path]):
     constraints_file: Optional[os.PathLike] = None
     python_version: Optional[str] = None
     extra_index_urls: List[str] = field(default_factory=list)
+    tags: List[str] = field(default_factory=list)
 
     @classmethod
     def from_config(
@@ -54,6 +55,7 @@ class VirtualPythonEnvironment(BaseEnvironment[Path]):
             *self.requirements,
             *constraints,
             *self.extra_index_urls,
+            *sorted(self.tags),
         )
 
     def install_requirements(self, path: Path) -> None:

isolate/connections/common.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import importlib
 import os
+from dataclasses import dataclass
 from contextlib import contextmanager
 from typing import TYPE_CHECKING, Any, Iterator, Optional, cast
 
@@ -21,9 +22,12 @@ if TYPE_CHECKING:
 AGENT_SIGNATURE = "IS_ISOLATE_AGENT"
 
 
+@dataclass
 class SerializationError(Exception):
     """An error that happened during the serialization process."""
 
+    message: str
+
 
 @contextmanager
 def _step(message: str) -> Iterator[None]:

isolate/connections/grpc/agent.py

@@ -142,14 +142,17 @@ class AgentServicer(definitions.AgentServicer):
         serialization_method: str,
         result: object,
         was_it_raised: bool,
-        stringized_tb: str,
+        stringized_tb: str | None,
     ) -> Generator[definitions.PartialRunResult, None, Any]:
         try:
             definition = serialize_object(serialization_method, result)
         except SerializationError:
-            yield from self.log(traceback.format_exc(), level=LogLevel.ERROR)
+            if stringized_tb:
+                yield from self.log(
+                    stringized_tb, source=LogSource.USER, level=LogLevel.STDERR
+                )
             raise AbortException(
-                "The result of the input function could not be serialized."
+                f"Error while serializing the execution result (object of type {type(result)})."
             )
         except BaseException:
             yield from self.log(traceback.format_exc(), level=LogLevel.ERROR)

isolate/server/server.py

@@ -261,6 +261,17 @@ class IsolateServicer(definitions.IsolateServicer):
                 # during the execution, and handle them accordingly.
                 exception = future.exception(timeout=0.1)
                 if exception is not None:
+                    # If this is an RPC error, propagate it as is without any
+                    # further processing.
+                    if isinstance(exception, grpc.RpcError):
+                        return self.abort_with_msg(
+                            exception.details(),
+                            context,
+                            code=exception.code(),
+                        )
+
+                    # Otherwise this is a bug in the agent itself, so needs
+                    # to be propagated with more details.
                     for line in traceback.format_exception(
                         type(exception), exception, exception.__traceback__
                     ):
@@ -282,7 +293,8 @@ class IsolateServicer(definitions.IsolateServicer):
         self, queue: Queue, is_completed: Callable[[], bool]
     ) -> Iterator[definitions.PartialRunResult]:
         """Watch the given queue until the is_completed function returns True. Note that even
-        if the function is completed, this function might not finish until the queue is empty."""
+        if the function is completed, this function might not finish until the queue is empty.
+        """
         while not is_completed():
             try:
                 yield queue.get(timeout=_Q_WAIT_DELAY)

{isolate-0.12.0.dist-info → isolate-0.12.1.dist-info}/METADATA

@@ -1,25 +1,26 @@
 Metadata-Version: 2.1
 Name: isolate
-Version: 0.12.0
+Version: 0.12.1
 Summary: Managed isolated environments for Python
 Author: Features & Labels
 Author-email: hello@fal.ai
 Requires-Python: >=3.7,<4.0
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.7
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
 Provides-Extra: build
 Provides-Extra: grpc
 Provides-Extra: server
-Requires-Dist: PyYAML (>=6.0); extra == "build"
+Requires-Dist: PyYAML (>=6.0) ; extra == "build"
 Requires-Dist: grpcio (>=1.49)
-Requires-Dist: importlib-metadata (>=4.4); python_version < "3.10"
+Requires-Dist: importlib-metadata (>=4.4) ; python_version < "3.10"
 Requires-Dist: platformdirs
 Requires-Dist: protobuf
 Requires-Dist: tblib (>=1.7.0)
-Requires-Dist: virtualenv (>=20.4); extra == "build"
+Requires-Dist: virtualenv (>=20.4) ; extra == "build"
 Description-Content-Type: text/markdown
 
 # Isolate

{isolate-0.12.0.dist-info → isolate-0.12.1.dist-info}/RECORD

@@ -2,22 +2,22 @@ isolate/__init__.py,sha256=zTNX62myAQBEKk3BUQ9_-eYryNzEVPpuWRbqkVyCsCo,49
 isolate/backends/__init__.py,sha256=_Z92FgH02UYPlF5azuzb2MZUJcmSzLOO0Ry__VX2CWE,91
 isolate/backends/_base.py,sha256=WicmqNs1N5Tof4XAjveyt3FxR2qe-9ymE5J_B3-xqpY,4131
 isolate/backends/common.py,sha256=TR3sLs_AzWA70IMPHpInxHc9pAqVhNWgpqCJhMfiKKE,7779
-isolate/backends/conda.py,sha256=sqtwfSki7EkTfADWkXGjy995dm8x7fzir4DtpJ3C950,7117
+isolate/backends/conda.py,sha256=Hvb9C6jMkvTqwdi_a7mWxwdnFqxCqBWPY-uMgSXLDU0,7255
 isolate/backends/local.py,sha256=n_RKUHsEEUo1oyLlRw0PzUAbwDC9qMjp9SvnDFeDFE0,1366
 isolate/backends/pyenv.py,sha256=AoWbphdZpxpoo704J2tS4l-gfzMrPcUD1gm1ORMhUek,5404
 isolate/backends/remote.py,sha256=2JA5RZISb4Qa_8a8aYnQrinpDZG89dtUnSRvJeTAx78,4208
 isolate/backends/settings.py,sha256=IT32xZuT3S2FbxfY1EdC6f0DbE0TKlhfPMWgoe4jBSI,3084
-isolate/backends/virtualenv.py,sha256=EJnjeUW3GvkIYX3FLJNmhNcVxDWJUBG0wXZ1oOpz-1o,5520
+isolate/backends/virtualenv.py,sha256=M6p-E8P5I7ed_k3lfUPiPhnE3k02o1B9-N0XFkix5FE,5602
 isolate/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 isolate/common/timestamp.py,sha256=seh7FrMRH4i1SCQavA8d-7z8qi0pP8lYYhd29gTPMwE,367
 isolate/connections/__init__.py,sha256=6BKY7P23ZZ2au0Efc2RO7Pt-kVqYF-Qprrz-rLIfyXc,722
 isolate/connections/_local/__init__.py,sha256=Zklry8jiqMw6-MTfS1L6pqmCCqRzOOGCyZgVLGRpDVY,118
 isolate/connections/_local/_base.py,sha256=2R5uenG-OAOOTmgVoADs7xoSCemwTlWsIpYRKVNYFu0,6251
 isolate/connections/_local/agent_startup.py,sha256=swCs6Q0yVkDw7w-RftizHSMyJDM7DQwuP3TB0qI1ucg,1552
-isolate/connections/common.py,sha256=gKvafqwZmGLhVbm-vlMJD2hOQUL7GJYOUwk3KTa0HMU,3475
+isolate/connections/common.py,sha256=dZzQrDpECQgj8xTuqmkv62BY6QSSumwrM-hWlWHadyE,3538
 isolate/connections/grpc/__init__.py,sha256=jQ9oN8X2o64pm0lSwc1fin_45ZJkFe1-i5zq8PdX2CU,71
 isolate/connections/grpc/_base.py,sha256=f2OpaeEaHCTnLPA3jcWHYDLPEuqX14otPd82UT5hdz8,5529
-isolate/connections/grpc/agent.py,sha256=L4_WVcyIdFeMDZNPQhweoYKJ3nIdsUfIJgxlZW7qHf4,7677
+isolate/connections/grpc/agent.py,sha256=KpAyt4p6glfiJaCJKElm3JT5uZcvHauNphabMYKIJek,7792
 isolate/connections/grpc/configuration.py,sha256=50YvGGHA9uyKg74xU_gc73j7bsFk973uIpMhmw2HhxY,788
 isolate/connections/grpc/definitions/__init__.py,sha256=-A4fhg-y-IuXTMsZsa8Eyg1V0mvctoh1s4Igv8Fexa0,389
 isolate/connections/grpc/definitions/agent.proto,sha256=Hx11hHc8PKwhWzyasViLeq7JL33KsRex2-iibfWruTw,568
@@ -47,8 +47,8 @@ isolate/server/health/health_pb2.pyi,sha256=boMRHMlX770EuccQCFTeRgf_KA_VMgW7l9GZ
 isolate/server/health/health_pb2_grpc.py,sha256=JRluct2W4af83OYxwmcCn0vRc78zf04Num0vBApuPEo,4005
 isolate/server/health_server.py,sha256=yN7F1Q28DdX8-Zk3gef7XcQEE25XwlHwzV5GBM75aQM,1249
 isolate/server/interface.py,sha256=nGbjdxrN0p9m1LNdeds8NIoJOwPYW2NM6ktmbhfG4_s,687
-isolate/server/server.py,sha256=bUxU_iohlBaBVYPvh-QDQOtEVAYCUP1IyeC9byzPuuY,12414
-isolate-0.12.0.dist-info/entry_points.txt,sha256=QXWwDC7bzMidCWvv7WrIKvlWneFKA21c3SDMVvgHpT4,281
-isolate-0.12.0.dist-info/WHEEL,sha256=DA86_h4QwwzGeRoz62o1svYt5kGEXpoUTuTtwzoTb30,83
-isolate-0.12.0.dist-info/METADATA,sha256=8Ou44xJde0uJqYKP1tww5O5xnTE0UCMNxA8iHDWvCBY,2771
-isolate-0.12.0.dist-info/RECORD,,
+isolate/server/server.py,sha256=XemBO-cdWNquM3cW5Sh-uc5h_sLgipo-x4DExpAfEzs,12954
+isolate-0.12.1.dist-info/METADATA,sha256=vXkwKv9COm7Ca-9oxB7qX7g_jvtdEIFcYM5OCKz72i8,2825
+isolate-0.12.1.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+isolate-0.12.1.dist-info/entry_points.txt,sha256=QXWwDC7bzMidCWvv7WrIKvlWneFKA21c3SDMVvgHpT4,281
+isolate-0.12.1.dist-info/RECORD,,

{isolate-0.12.0.dist-info → isolate-0.12.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry 1.0.8
+Generator: poetry-core 1.5.2
 Root-Is-Purelib: true
 Tag: py3-none-any