truefoundry 0.4.0rc3__tar.gz → 0.4.1__tar.gz

{truefoundry-0.4.0rc3 → truefoundry-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.0rc3
+Version: 0.4.1
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com
@@ -45,7 +45,7 @@ Requires-Dist: rich-click (>=1.2.1,<2.0.0)
 Requires-Dist: scipy (>=1.12.0,<2.0.0) ; python_version >= "3.12"
 Requires-Dist: scipy (>=1.5.0,<2.0.0) ; python_version < "3.12"
 Requires-Dist: tqdm (>=4.0.0,<5.0.0)
-Requires-Dist: typing-extensions (>=3.10.0)
+Requires-Dist: typing-extensions (>=4.0)
 Requires-Dist: urllib3 (>=1.26.18,<3)
 Requires-Dist: uvicorn (>=0.13.0,<1.0.0)
 Requires-Dist: yq (>=3.1.0,<4.0.0)
@@ -68,10 +68,10 @@ To install the `TrueFoundry CLI`, you can use pip:
 pip install truefoundry
 ```
 
-To install the `TrueFoundry CLI` with metadata/artifact tracking, you can use pip:
+To install the `TrueFoundry CLI` with Workflow deployment features, you can use pip:
 
 ```bash
-pip install truefoundry[ml]
+pip install truefoundry[workflow]
 ```
 
 ## Examples

{truefoundry-0.4.0rc3 → truefoundry-0.4.1}/README.md

@@ -15,10 +15,10 @@ To install the `TrueFoundry CLI`, you can use pip:
 pip install truefoundry
 ```
 
-To install the `TrueFoundry CLI` with metadata/artifact tracking, you can use pip:
+To install the `TrueFoundry CLI` with Workflow deployment features, you can use pip:
 
 ```bash
-pip install truefoundry[ml]
+pip install truefoundry[workflow]
 ```
 
 ## Examples

{truefoundry-0.4.0rc3 → truefoundry-0.4.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.4.0rc3"
+version = "0.4.1"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"
@@ -35,7 +35,7 @@ requirements-parser = ">=0.11.0,<0.12.0"
 rich = ">=13.7.1,<14.0.0"
 rich-click = ">=1.2.1,<2.0.0"
 tqdm = ">=4.0.0,<5.0.0"
-typing-extensions = ">=3.10.0"
+typing-extensions = ">=4.0"
 urllib3 = ">=1.26.18,<3"
 uvicorn = ">=0.13.0,<1.0.0"
 yq = ">=3.1.0,<4.0.0"
@@ -77,16 +77,6 @@ shap =  [
     { version = ">=0.40.0", python = "<3.12" },
     { version = ">=0.45.0", python = ">=3.12" },
 ]
-pandas-stubs = "*"
-types-Pygments = "*"
-types-cffi = "*"
-types-colorama = "*"
-types-decorator = "*"
-types-protobuf = "*"
-types-python-dateutil = "*"
-types-requests = "*"
-types-six = "*"
-types-tqdm = "*"
 
 
 [tool.poetry-dynamic-versioning]

{truefoundry-0.4.0rc3 → truefoundry-0.4.1}/truefoundry/common/entities.py

@@ -3,7 +3,7 @@ from enum import Enum
 from typing import Optional
 
 import jwt
-from typing_extensions import TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 from truefoundry.pydantic_v1 import BaseModel, Field, NonEmptyStr, validator
 
@@ -28,7 +28,7 @@ class _DecodedToken(TypedDict):
     tenantName: str
     exp: int
     username: str
-    email: str
+    email: NotRequired[str]
     userType: UserType
 
 

{truefoundry-0.4.0rc3 → truefoundry-0.4.1}/truefoundry/deploy/python_deploy_codegen.py

@@ -101,11 +101,11 @@ def add_local_source_comment(code):
     return "\n".join(new_lines)
 
 
-def convert_deployment_config_to_python(workspace_fqn: str, deployment_config: dict):
+def convert_deployment_config_to_python(workspace_fqn: str, application_spec: dict):
     """
     Convert a deployment config to a python file that can be used to deploy to a workspace
     """
-    application = Application.parse_obj(deployment_config)
+    application = Application.parse_obj(application_spec)
     application_type = application.__root__.type
 
     spec_repr = get_python_repr(application.__root__)

{truefoundry-0.4.0rc3 → truefoundry-0.4.1}/truefoundry/ml/artifact/truefoundry_artifact_repo.py

@@ -236,7 +236,7 @@ def _signed_url_upload_file(
         with cloud_storage_http_request(
             method="put", url=signed_url.signed_url, data=""
         ) as response:
-            augmented_raise_for_status(response.raise_for_status())
+            augmented_raise_for_status(response)
         return
 
     task_progress_bar = progress_bar.add_task(

{truefoundry-0.4.0rc3 → truefoundry-0.4.1}/truefoundry/ml/clients/utils.py

@@ -23,7 +23,7 @@ def _http_request(
     headers = kwargs.pop("headers", {}) or {}
     if token is not None:
         headers["Authorization"] = f"Bearer {token}"
-    return session.request(method=method, url=url, **kwargs)
+    return session.request(method=method, url=url, headers=headers, **kwargs)
 
 
 def http_request(