asta-code-execution 0.1.2__tar.gz → 0.1.4__tar.gz

{asta_code_execution-0.1.2 → asta_code_execution-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: asta-code-execution
-Version: 0.1.2
+Version: 0.1.4
 Summary: Add your description here
 Author: Allen Institute for Artificial Intelligence
 Requires-Dist: ipython>=9.9.0

{asta_code_execution-0.1.2 → asta_code_execution-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "asta-code-execution"
-version = "0.1.2"
+version = "0.1.4"
 description = "Add your description here"
 readme = "README.md"
 authors = [

{asta_code_execution-0.1.2 → asta_code_execution-0.1.4}/src/code_execution/process_backend.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import importlib.metadata
 import json
 import os
 import shutil
@@ -148,10 +149,20 @@ class ProcessIPythonBackend:
                 text=True,
             )
 
-        # Install code_execution package (editable) so the sandbox runner works
-        code_exec_pkg = Path(__file__).resolve().parent.parent.parent  # packages/code_execution
+        # Install code_execution package so the sandbox runner can import it.
+        # In the workspace dev layout, packages/code_execution/pyproject.toml is
+        # three levels up from this file; install editable from there. When the
+        # parent process is running from a normal site-packages install, that
+        # path is the site-packages parent dir (no pyproject.toml), so fall back
+        # to installing the published wheel pinned to the loaded version.
+        code_exec_pkg = Path(__file__).resolve().parent.parent.parent
+        if (code_exec_pkg / "pyproject.toml").is_file():
+            install_spec = ["-e", str(code_exec_pkg)]
+        else:
+            version = importlib.metadata.version("asta-code-execution")
+            install_spec = [f"asta-code-execution=={version}"]
         subprocess.run(
-            [uv, "pip", "install", "--python", str(python_bin), "-e", str(code_exec_pkg)],
+            [uv, "pip", "install", "--python", str(python_bin), *install_spec],
             check=True,
             capture_output=True,
             text=True,