kubetorch 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

kubetorch/__init__.py

@@ -57,4 +57,4 @@ EXCEPTION_REGISTRY = {
 for exception in EXCEPTION_REGISTRY.values():
     exception.__module__ = "kubetorch"
 
-__version__ = "0.2.0"
+__version__ = "0.2.2"

kubetorch/servers/http/http_server.py

@@ -1309,8 +1309,9 @@ async def lifespan(app: FastAPI):
         logger.debug("No TTL annotation found, heartbeat disabled")
 
     try:
-        cached_image_setup()
-        if not os.getenv("KT_CALLABLE_TYPE") == "app":
+        if os.getenv("KT_CALLABLE_TYPE") == "app":
+            cached_image_setup()
+        else:
             load_callable()
 
         logger.info("Kubetorch Server started.")

kubetorch/serving/templates/kt_setup_template.sh.j2

@@ -59,12 +59,13 @@ fi
 # Export the install command as an environment variable for use in applications
 echo "Setting KT_PIP_INSTALL_CMD env var to $uv_pip_cmd"
 export KT_PIP_INSTALL_CMD="$uv_pip_cmd"
+mkdir -p .kt
 echo "$uv_pip_cmd" > .kt/kt_pip_install_cmd
 
 {% if install_url %}
-  $uv_pip_cmd "kubetorch[server] @ {{ install_url }}"
+  $uv_pip_cmd "kubetorch[server]=={{ install_url }}"
   {% if install_otel %}
-    $uv_pip_cmd "kubetorch[otel] @ {{ install_url }}"
+    $uv_pip_cmd "kubetorch[otel]=={{ install_url }}"
   {% endif %}
 {% else %}
   {{ rsync_kt_editable_cmd }}

kubetorch/utils.py

@@ -1,5 +1,4 @@
 import enum
-import importlib.util
 import json
 import os
 import re
@@ -123,30 +122,10 @@ def get_kt_install_url(freeze: bool = False):
     if local_kt_path and (Path(local_kt_path) / "pyproject.toml").exists():
         return local_kt_path, True
     else:
-        # If the user is using uv, sometimes the pip freeze command won't return the full "kubetorch @ ..." install
-        # URL. However, running `uv pip freeze` will return the full URL, even when the user isn't actually in a uv
-        # venv. So we default to using `uv pip freeze` if uv is installed, otherwise we use the regular `pip freeze`.
-        uv_installed = importlib.util.find_spec("uv") is not None
-        # check if installed from presigned url
-        freeze_cmd = (
-            "pip freeze | grep kubetorch"
-            if not uv_installed
-            else "uv pip freeze | grep kubetorch"
-        )
-        output = subprocess.run(
-            freeze_cmd,
-            shell=True,
-            capture_output=True,
-            text=True,
-        ).stdout
-        if not output.startswith("kubetorch @ "):
-            raise Exception(
-                "Could not find kubetorch version to install on the pod. You must either set the "
-                "``install_url`` in the config or ``KT_INSTALL_URL`` env var, have kubetorch installed "
-                "locally, or set ``compute.freeze`` to ``True``."
-            )
-        install_url = output[len("kubetorch @ ") :].strip()
-        return install_url, False
+        import kubetorch as kt
+
+        version = kt.__version__
+        return version, False
 
 
 class LogVerbosity(str, enum.Enum):

{kubetorch-0.2.0.dist-info → kubetorch-0.2.2.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: kubetorch
-Version: 0.2.0
-Summary: Kubetorch: A Kubernetes-native framework for distributed PyTorch workloads
+Version: 0.2.2
+Summary: A Python interface for running ML workloads on Kubernetes
 License: Apache 2.0
 Author: Runhouse Team
 Requires-Python: >=3.9,<=3.13
@@ -97,10 +97,10 @@ pip install "kubetorch[client]"
 ```bash
 # Option 1: Install directly from OCI registry
 helm upgrade --install kubetorch oci://ghcr.io/run-house/charts/kubetorch \
-  --version 0.2.0 -n kubetorch --create-namespace
+  --version 0.2.2 -n kubetorch --create-namespace
 
 # Option 2: Download chart locally first
-helm pull oci://ghcr.io/run-house/charts/kubetorch --version 0.2.0 --untar
+helm pull oci://ghcr.io/run-house/charts/kubetorch --version 0.2.2 --untar
 helm upgrade --install kubetorch ./kubetorch -n kubetorch --create-namespace
 ```
 

{kubetorch-0.2.0.dist-info → kubetorch-0.2.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kubetorch/__init__.py,sha256=EwSzER4seuElVH9dpIu9k09dHfvnfBx1dp_zu-0B4h0,2115
+kubetorch/__init__.py,sha256=JZaiU7VcuwqRpmzdhxJaXt9VbQtVagJ8gp4dMnPs8BU,2115
 kubetorch/cli.py,sha256=yCn-1z71TNS-tvA2D5CbZF3YTh2QYvfyXi7H3yN2vXI,68611
 kubetorch/cli_utils.py,sha256=zzABvAT35eQulAakAmo6PwZkMg6pZNLA3HoZqsYZQB0,33353
 kubetorch/config.py,sha256=-up9Tljkls_ZU8bASjDRnFaxTknBsoLeWgbuXbSt6cw,15839
@@ -66,7 +66,7 @@ kubetorch/servers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 kubetorch/servers/http/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kubetorch/servers/http/distributed_utils.py,sha256=6CLZn4JOdTIWSvnN5DYOIhiT7NctFeEtL-nqOAx4bYM,129685
 kubetorch/servers/http/http_client.py,sha256=xs84zNGnquiF_q3FNcdGEKXu9LLmrnKxeJcAlZzP27I,29753
-kubetorch/servers/http/http_server.py,sha256=CscBJBd7GiHxuWTopvK0JKFqqy7fyjM_aLUVEGVCJ30,69822
+kubetorch/servers/http/http_server.py,sha256=y0AO5LLPozK1mDDrbWvW9sVwqxwSKZppWN7io7FAp5A,69836
 kubetorch/servers/http/server_metrics.py,sha256=0c9K7DBo6fYXAYqCXf2zqDyARP7mB2fwPOuDTzVTIQk,9614
 kubetorch/servers/http/utils.py,sha256=39MQDD027D5YwA78U_hBVjbSvEsymYwZ-SVvQ4Zpzk8,27231
 kubetorch/serving/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -79,15 +79,15 @@ kubetorch/serving/raycluster_service_manager.py,sha256=k3u01KDsi8tW_NWzR3jytT2DA
 kubetorch/serving/service_manager.py,sha256=4LdrzfwPFPf7IsSwR7HVxuSrbyBNw1dI4xPHR8TECrw,720
 kubetorch/serving/templates/deployment_template.yaml,sha256=_hPm7sXCb2BVeYuhyAGz-hRzjAoSaXNikJj_3MJfLsM,435
 kubetorch/serving/templates/knative_service_template.yaml,sha256=yKxBdduIOqDntywBdUFVaV3hc50cZhl6K91FNok6X_U,553
-kubetorch/serving/templates/kt_setup_template.sh.j2,sha256=_U5TAq7WW0ALORmKS5Ma9R8e0TUthVh3PbQi21YxryU,2680
+kubetorch/serving/templates/kt_setup_template.sh.j2,sha256=OqTWkMmiqDodia0a9f-wnef1EBL4XFWBWuXdbpotVKc,2691
 kubetorch/serving/templates/pod_template.yaml,sha256=5N0KuLBtphA1dyZ2hq8nYKOpaqDxUvLkZEl2fPG89yw,5530
 kubetorch/serving/templates/raycluster_service_template.yaml,sha256=kW_bIJu36Ud4kI8Q9ycnQfIIbhkxNa-C6OP93Jpl-EA,1037
 kubetorch/serving/templates/raycluster_template.yaml,sha256=ruWFk09-E85WZJc64l7NXrsdGYcW4spPzErWuzrLLes,990
 kubetorch/serving/templates/service_template.yaml,sha256=ClIMlW-JzK4I6YWo9oBbwYXgNTcz8AGDMpo8_Q8C4ng,642
 kubetorch/serving/templates/workerset_template.yaml,sha256=l7svttZrQPHDbZuZsO8EmG38eVlFUDr_UyP3E2Ci0zY,981
 kubetorch/serving/utils.py,sha256=7TASPzrK-8bX3lBvGj95I3fTp4MOBFqFhJvqA1TVvqA,12691
-kubetorch/utils.py,sha256=NIuWEWuBzOQsXb_eI5vEVYE_bu9uFzYVlyhHPQQSggo,9742
-kubetorch-0.2.0.dist-info/METADATA,sha256=IqB_-YH88RVJKCJkIDL0NFnetEDO-XxVKgm3I9HRIKI,4827
-kubetorch-0.2.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-kubetorch-0.2.0.dist-info/entry_points.txt,sha256=Gkg5B9pHJpGf-Zxk1LGx7RIG_9EI04MAbD6m4l1Rx3U,107
-kubetorch-0.2.0.dist-info/RECORD,,
+kubetorch/utils.py,sha256=h7bqR0OlmyTH61lKHGd6jEPNlBw8eX_5r0dN99ha_ws,8583
+kubetorch-0.2.2.dist-info/METADATA,sha256=DpgtgZVM8VxcNajYjSBYcUQ-iLNtz0Z8zTESwdlvHLw,4810
+kubetorch-0.2.2.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+kubetorch-0.2.2.dist-info/entry_points.txt,sha256=Gkg5B9pHJpGf-Zxk1LGx7RIG_9EI04MAbD6m4l1Rx3U,107
+kubetorch-0.2.2.dist-info/RECORD,,