kubetorch 0.2.0__tar.gz → 0.2.1__tar.gz

{kubetorch-0.2.0 → kubetorch-0.2.1}/PKG-INFO

@@ -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.1
+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.1 -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.1 --untar
 helm upgrade --install kubetorch ./kubetorch -n kubetorch --create-namespace
 ```
 

{kubetorch-0.2.0 → kubetorch-0.2.1}/README.md

@@ -43,10 +43,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.1 -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.1 --untar
 helm upgrade --install kubetorch ./kubetorch -n kubetorch --create-namespace
 ```
 

{kubetorch-0.2.0 → kubetorch-0.2.1}/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.1"

{kubetorch-0.2.0 → kubetorch-0.2.1}/kubetorch/serving/templates/kt_setup_template.sh.j2

@@ -62,9 +62,9 @@ export KT_PIP_INSTALL_CMD="$uv_pip_cmd"
 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-0.2.0 → kubetorch-0.2.1}/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 → kubetorch-0.2.1}/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "kubetorch"
-version = "0.2.0"
-description = "Kubetorch: A Kubernetes-native framework for distributed PyTorch workloads"
+version = "0.2.1"
+description = "A Python interface for running ML workloads on Kubernetes"
 authors = ["Runhouse Team"]
 license = "Apache 2.0"
 readme = "README.md"