truefoundry 0.5.7rc1__py3-none-any.whl → 0.5.8rc2__py3-none-any.whl

truefoundry/cli/__main__.py

@@ -71,7 +71,7 @@ def truefoundry_cli(ctx, json, debug):
     logger.add_cli_handler(level=log_level)
 
 
-def create_truefoundry_cli() -> click.MultiCommand:
+def create_truefoundry_cli() -> click.Group:
     """Generates CLI by combining all subcommands into a main CLI and returns in
 
     Returns:
@@ -105,7 +105,7 @@ def main():
     try:
         cli = create_truefoundry_cli()
     except Exception as e:
-        raise click.exceptions.UsageError(message=str(e)) from e
+        raise click.UsageError(message=str(e)) from e
     sys.exit(cli())
 
 

truefoundry/common/credential_provider.py

@@ -38,7 +38,7 @@ class EnvCredentialProvider(CredentialProvider):
             )
         self._host = resolve_tfy_host()
         self._auth_service = AuthServiceClient.from_tfy_host(tfy_host=self._host)
-        self._token: Token = Token(access_token=api_key, refresh_token=None)  # type: ignore[call-arg]
+        self._token: Token = Token(access_token=api_key, refresh_token=None)
 
     @staticmethod
     def can_provide() -> bool:

truefoundry/common/entities.py

@@ -62,7 +62,7 @@ class Token(BaseModel):
         return (exp - time.time()) < buffer_in_seconds
 
     def to_user_info(self) -> UserInfo:
-        return UserInfo(  # type: ignore[call-arg]
+        return UserInfo(
             user_id=self.decoded_value["username"],
             email=self.decoded_value["email"]
             if "email" in self.decoded_value
@@ -81,9 +81,7 @@ class CredentialsFileContent(BaseModel):
     host: NonEmptyStr
 
     def to_token(self) -> Token:
-        return Token(  # type: ignore[call-arg]
-            access_token=self.access_token, refresh_token=self.refresh_token
-        )
+        return Token(access_token=self.access_token, refresh_token=self.refresh_token)
 
 
 class TenantInfo(BaseModel):

truefoundry/common/servicefoundry_client.py

@@ -1,9 +1,11 @@
 from __future__ import annotations
 
 import functools
+from typing import Callable, TypeVar
 
 import requests
 from packaging import version
+from typing_extensions import ParamSpec
 
 from truefoundry.common.constants import (
     SERVICEFOUNDRY_CLIENT_MAX_RETRIES,
@@ -24,10 +26,13 @@ from truefoundry.common.utils import (
 from truefoundry.logger import logger
 from truefoundry.version import __version__
 
+P = ParamSpec("P")
+R = TypeVar("R")
 
-def check_min_cli_version(fn):
+
+def check_min_cli_version(fn: Callable[P, R]) -> Callable[P, R]:
     @functools.wraps(fn)
-    def inner(*args, **kwargs):
+    def inner(*args: P.args, **kwargs: P.kwargs) -> R:
         client: "ServiceFoundryServiceClient" = args[0]
         client.check_min_cli_version()
         return fn(*args, **kwargs)

truefoundry/common/types.py

@@ -0,0 +1,7 @@
+from typing import Protocol
+
+
+class UploadCodePackageCallable(Protocol):
+    def __call__(
+        self, workspace_fqn: str, component_name: str, package_local_path: str
+    ) -> str: ...

truefoundry/common/utils.py

@@ -53,7 +53,7 @@ def get_tfy_servers_config(tfy_host: str) -> _TFYServersConfig:
     global _tfy_servers_config
     if _tfy_servers_config is None:
         if ENV_VARS.TFY_CLI_LOCAL_DEV_MODE:
-            _tfy_servers_config = _TFYServersConfig()  # type: ignore[call-arg]
+            _tfy_servers_config = _TFYServersConfig()
         else:
             _tfy_servers_config = _TFYServersConfig.from_tfy_host(tfy_host)
     return _tfy_servers_config

truefoundry/deploy/builder/docker_service.py

@@ -38,7 +38,7 @@ def _catch_error_in_push(response: List[dict]):
     for line in response:
         if line.get("error") is not None:
             raise Exception(
-                f'Failed to push to registry with message \'{line.get("error")}\''
+                f"Failed to push to registry with message '{line.get('error')}'"
             )
 
 

truefoundry/deploy/cli/commands/deploy_command.py

@@ -55,7 +55,7 @@ def _get_default_spec_file():
     help="Wait and tail the deployment progress",
 )
 @click.option(
-    "--force-deploy/--no-force-deploy",
+    "--force/--no-force",
     is_flag=True,
     show_default=True,
     default=False,
@@ -68,7 +68,7 @@ def deploy_command(
     file: str,
     workspace_fqn: Optional[str],
     wait: bool,
-    force_deploy: bool,
+    force: bool = False,
 ):
     if ctx.invoked_subcommand is not None:
         return
@@ -85,9 +85,7 @@ def deploy_command(
             application_definition = yaml.safe_load(f)
 
         application = Application.parse_obj(application_definition)
-        application.deploy(
-            workspace_fqn=workspace_fqn, wait=wait, force_deploy=force_deploy
-        )
+        application.deploy(workspace_fqn=workspace_fqn, wait=wait, force=force)
         sys.exit(0)
 
     click.echo(

truefoundry/deploy/lib/clients/servicefoundry_client.py

@@ -4,7 +4,7 @@ import json
 import os
 import time
 from datetime import datetime, timezone
-from typing import TYPE_CHECKING, Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional
 from urllib.parse import urljoin
 
 import requests
@@ -50,9 +50,6 @@ DEPLOYMENT_LOGS_SUBSCRIBE_MESSAGE = "DEPLOYMENT_LOGS"
 BUILD_LOGS_SUBSCRIBE_MESSAGE = "BUILD_LOGS"
 MAX_RETRIES_WORKFLOW_TRIGGER = 3
 
-if TYPE_CHECKING:
-    from truefoundry.deploy.auto_gen.models import Application
-
 
 def _upload_packaged_code(metadata, package_file):
     file_size = os.stat(package_file).st_size
@@ -242,13 +239,13 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
         self,
         workspace_id: str,
         application: auto_gen_models.Workflow,
-        force_deploy: bool,
+        force: bool = False,
     ) -> Deployment:
         data = {
             "workspaceId": workspace_id,
             "name": application.name,
             "manifest": application.dict(exclude_none=True),
-            "forceDeploy": force_deploy,
+            "forceDeploy": force,
         }
         logger.debug(json.dumps(data))
         url = f"{self._api_server_url}/{VERSION_PREFIX}/deployment"
@@ -265,7 +262,7 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
         time_obj.replace(tzinfo=timezone.utc)
         local_time = time_obj.astimezone(tzlocal())
         local_time_str = local_time.isoformat()
-        return f'[{local_time_str}] {log["log"].strip()}'
+        return f"[{local_time_str}] {log['log'].strip()}"
 
     def _tail_logs(
         self,

truefoundry/deploy/v2/lib/deploy.py

@@ -214,8 +214,8 @@ No Workspace FQN was provided or mentioned in the spec.
 Either add a `workspace_fqn` to your yaml spec as
 
 ```
-name: {getattr(component, 'name', 'my-app')}
-type: {getattr(component, 'type', 'undefined')}
+name: {getattr(component, "name", "my-app")}
+type: {getattr(component, "type", "undefined")}
 ...
 workspace_fqn: <your workspace fqn>
 ```
@@ -224,7 +224,7 @@ or Python deployment spec as
 
 ```
 app = {component.__class__.__name__}(
-    name='{getattr(component, 'name', 'my-app')}',
+    name='{getattr(component, "name", "my-app")}',
     ...
     workspace_fqn='<your workspace fqn>'
 )
@@ -252,7 +252,7 @@ def deploy_component(
     component: Component,
     workspace_fqn: Optional[str] = None,
     wait: bool = True,
-    force_deploy: bool = False,
+    force: bool = False,
 ) -> Deployment:
     _warn_when_gpu_selected_without_cuda(component=component)
     workspace_fqn = _resolve_workspace_fqn(
@@ -274,7 +274,7 @@ def deploy_component(
     response = client.deploy_application(
         workspace_id=workspace_id,
         application=updated_component,
-        force_deploy=force_deploy,
+        force=force,
     )
     logger.info(
         "🚀 Deployment started for application '%s'. Deployment FQN is '%s'.",

truefoundry/deploy/v2/lib/deploy_workflow.py

@@ -1,7 +1,7 @@
 import os
 import sys
 from pathlib import Path
-from typing import Callable, Dict, List, Optional, Union
+from typing import Dict, List, Optional, Union
 
 import requirements
 from flytekit.configuration import (
@@ -19,6 +19,7 @@ from flytekit.tools.translator import TaskSpec as FlyteTaskSpec
 from flytekit.tools.translator import WorkflowSpec as FlyteWorkflowSpec
 from google.protobuf.json_format import MessageToDict
 
+from truefoundry.common.types import UploadCodePackageCallable
 from truefoundry.deploy.auto_gen import models as auto_gen_models
 from truefoundry.deploy.lib.clients.servicefoundry_client import (
     ServiceFoundryServiceClient,
@@ -39,7 +40,7 @@ from truefoundry.workflow.workflow import (
 def _handle_code_upload_for_workflow(
     workflow: auto_gen_models.Workflow,
     workspace_fqn: str,
-    upload_code_package: Callable[[str, str, str], str],
+    upload_code_package: UploadCodePackageCallable,
 ) -> auto_gen_models.Workflow:
     new_workflow = workflow.copy(deep=True)
     new_workflow.source = local_source_to_remote_source(
@@ -92,7 +93,7 @@ def _is_tfy_wf_present_in_task_python_build(
 
 
 def _is_dynamic_task(flyte_task: FlyteTaskSpec) -> bool:
-    envs: Dict[str:str] = flyte_task.template.container.env or {}
+    envs: Dict[str, str] = flyte_task.template.container.env or {}
     return SERIALIZED_CONTEXT_ENV_VAR in envs.keys()
 
 
@@ -273,7 +274,7 @@ def deploy_workflow(
     workflow: auto_gen_models.Workflow,
     workspace_fqn: str,
     wait: bool = True,
-    force_deploy: bool = False,
+    force: bool = False,
 ) -> Deployment:
     _generate_manifest_for_workflow(workflow)
     _validate_workspace_fqn(workflow, workspace_fqn)
@@ -296,7 +297,9 @@ def deploy_workflow(
     )
 
     deployment = client.deploy_application(
-        workspace_id=workspace_id, application=workflow
+        workspace_id=workspace_id,
+        application=workflow,
+        force=force,
     )
     logger.info(
         "🚀 Deployment started for application '%s'. Deployment FQN is '%s'.",

truefoundry/deploy/v2/lib/deployable_patched_models.py

@@ -12,13 +12,13 @@ class DeployablePatchedModelBase(BaseModel):
         extra = "forbid"
 
     def deploy(
-        self, workspace_fqn: str, wait: bool = True, force_deploy: bool = False
+        self, workspace_fqn: str, wait: bool = True, force: bool = False
     ) -> Deployment:
         return deploy_component(
             component=self,
             workspace_fqn=workspace_fqn,
             wait=wait,
-            force_deploy=force_deploy,
+            force=force,
         )
 
 
@@ -26,7 +26,7 @@ class Service(models.Service, DeployablePatchedModelBase):
     type: Literal["service"] = "service"
     resources: models.Resources = Field(default_factory=models.Resources)
     # This is being patched because cue export marks this as a "number"
-    replicas: Union[conint(ge=0, le=100), models.ServiceAutoscaling] = Field(
+    replicas: Union[conint(ge=0, le=100), models.ServiceAutoscaling] = Field(  # type: ignore[valid-type]
         1,
         description="+label=Replicas\n+usage=Replicas of service you want to run\n+icon=fa-clone\n+sort=3",
     )
@@ -70,7 +70,7 @@ class ApplicationSet(models.ApplicationSet, DeployablePatchedModelBase):
 
 class AsyncService(models.AsyncService, DeployablePatchedModelBase):
     type: Literal["async-service"] = "async-service"
-    replicas: Union[conint(ge=0, le=100), models.AsyncServiceAutoscaling] = 1
+    replicas: Union[conint(ge=0, le=100), models.AsyncServiceAutoscaling] = 1  # type: ignore[valid-type]
     resources: models.Resources = Field(default_factory=models.Resources)
 
 
@@ -81,7 +81,7 @@ class SSHServer(models.SSHServer, DeployablePatchedModelBase):
 
 class Application(models.Application, DeployablePatchedModelBase):
     def deploy(
-        self, workspace_fqn: str, wait: bool = True, force_deploy: bool = False
+        self, workspace_fqn: str, wait: bool = True, force: bool = False
     ) -> Deployment:
         if isinstance(self.__root__, models.Workflow):
             from truefoundry.deploy.v2.lib.deploy_workflow import deploy_workflow
@@ -90,14 +90,14 @@ class Application(models.Application, DeployablePatchedModelBase):
                 workflow=self.__root__,
                 workspace_fqn=workspace_fqn,
                 wait=wait,
-                force_deploy=force_deploy,
+                force=force,
             )
         else:
             return deploy_component(
                 component=self.__root__,
                 workspace_fqn=workspace_fqn,
                 wait=wait,
-                force_deploy=force_deploy,
+                force=force,
             )
 
 
@@ -107,7 +107,11 @@ class Workflow(models.Workflow, DeployablePatchedModelBase):
         default_factory=lambda: LocalSource(local_build=False)
     )
 
-    def deploy(self, workspace_fqn: str, wait: bool = True) -> Deployment:
+    def deploy(
+        self, workspace_fqn: str, wait: bool = True, force: bool = False
+    ) -> Deployment:
         from truefoundry.deploy.v2.lib.deploy_workflow import deploy_workflow
 
-        return deploy_workflow(workflow=self, workspace_fqn=workspace_fqn, wait=wait)
+        return deploy_workflow(
+            workflow=self, workspace_fqn=workspace_fqn, wait=wait, force=force
+        )

truefoundry/deploy/v2/lib/models.py

@@ -26,7 +26,7 @@ class BuildResponse(BaseModel):
 
 
 class AppDeploymentStatusResponse(BaseModel):
-    state: create_model(
+    state: create_model(  # type: ignore[valid-type]
         "State",
         isTerminalState=(bool, ...),
         type=(str, ...),

truefoundry/deploy/v2/lib/patched_models.py

@@ -518,7 +518,7 @@ class ArtifactsDownload(models.ArtifactsDownload, PatchedModelBase):
 
 class NvidiaGPU(models.NvidiaGPU, PatchedModelBase):
     type: Literal["nvidia_gpu"] = "nvidia_gpu"
-    name: Optional[Union[GPUType, constr(regex=r"^tpu-[a-z\d\-]+$")]] = None
+    name: Optional[Union[GPUType, constr(regex=r"^tpu-[a-z\d\-]+$")]] = None  # type: ignore[valid-type]
 
 
 class NvidiaMIGGPU(models.NvidiaMIGGPU, PatchedModelBase):

truefoundry/deploy/v2/lib/source.py

@@ -8,6 +8,7 @@ from typing import Callable, List, Optional
 import gitignorefile
 from tqdm import tqdm
 
+from truefoundry.common.types import UploadCodePackageCallable
 from truefoundry.common.warnings import TrueFoundryDeprecationWarning
 from truefoundry.deploy import builder
 from truefoundry.deploy.auto_gen import models
@@ -136,7 +137,7 @@ def local_source_to_remote_source(
     local_source: models.LocalSource,
     workspace_fqn: str,
     component_name: str,
-    upload_code_package: Callable[[str, str, str], str],
+    upload_code_package: UploadCodePackageCallable,
 ) -> RemoteSource:
     with tempfile.TemporaryDirectory() as local_dir:
         package_local_path = os.path.join(local_dir, "build.tar.gz")

truefoundry/ml/log_types/plot.py

@@ -33,7 +33,7 @@ if TYPE_CHECKING:
 PlotObjType = Union[
     "matplotlib.figure.Figure",
     "plotly.graph_objects.Figure",
-    "matplotlib.pyplot",
+    "matplotlib.pyplot",  # type: ignore[valid-type]
 ]
 
 
@@ -103,7 +103,7 @@ def _save_matplotlib_figure(
 
 
 def _save_matplotlib_plt(
-    plt: "matplotlib.pyplot",
+    plt: "matplotlib.pyplot",  # type: ignore[valid-type]
     key: str,
     step: int,
     local_dir: str,

truefoundry/ml/mlfoundry_run.py

@@ -1109,7 +1109,7 @@ class MlFoundryRun:
     @_ensure_not_deleted
     def log_plots(
         self,
-        plots: Dict[
+        plots: Dict[  # type: ignore[valid-type]
             str,
             Union[
                 "matplotlib.pyplot",

truefoundry-0.5.8rc2.dist-info/METADATA

@@ -0,0 +1,72 @@
+Metadata-Version: 2.4
+Name: truefoundry
+Version: 0.5.8rc2
+Summary: TrueFoundry CLI
+Author-email: TrueFoundry Team <abhishek@truefoundry.com>
+Requires-Python: <3.13,>=3.8.1
+Requires-Dist: aenum<4.0.0,>=3.0.0
+Requires-Dist: click<9.0.0,>=7.0.0
+Requires-Dist: coolname<2.0.0,>=1.1.0
+Requires-Dist: docker<8.0.0,>=6.1.2
+Requires-Dist: filelock<4.0.0,>=3.8.0
+Requires-Dist: gitignorefile<2.0.0,>=1.1.2
+Requires-Dist: gitpython<4.0.0,>=3.1.43
+Requires-Dist: importlib-metadata<9.0.0,>=4.11.3
+Requires-Dist: importlib-resources<7.0.0,>=5.2.0
+Requires-Dist: mako<2.0.0,>=1.1.6
+Requires-Dist: numpy<2.0.0,>=1.23.0; python_version < '3.12'
+Requires-Dist: numpy<2.0.0,>=1.26.0; python_version >= '3.12'
+Requires-Dist: openai<2.0.0,>=1.16.2
+Requires-Dist: packaging<25.0,>=20.0
+Requires-Dist: pydantic<3.0.0,>=1.8.2
+Requires-Dist: pygments<3.0.0,>=2.12.0
+Requires-Dist: pyjwt<3.0.0,>=2.0.0
+Requires-Dist: python-dateutil<3.0.0,>=2.8.2
+Requires-Dist: python-dotenv<2.0.0,>=1.0.1
+Requires-Dist: python-socketio[client]<6.0.0,>=5.5.2
+Requires-Dist: pyyaml<7.0.0,>=6.0.0
+Requires-Dist: questionary<2.0.0,>=1.10.0
+Requires-Dist: requests<3.0.0,>=2.18.0
+Requires-Dist: requirements-parser<0.12.0,>=0.11.0
+Requires-Dist: rich-click<2.0.0,>=1.2.1
+Requires-Dist: rich<14.0.0,>=13.7.1
+Requires-Dist: scipy<2.0.0,>=1.12.0; python_version >= '3.12'
+Requires-Dist: scipy<2.0.0,>=1.5.0; python_version < '3.12'
+Requires-Dist: tqdm<5.0.0,>=4.0.0
+Requires-Dist: typing-extensions>=4.0
+Requires-Dist: urllib3<3,>=1.26.18
+Requires-Dist: yq<4.0.0,>=3.1.0
+Provides-Extra: workflow
+Requires-Dist: flytekit==1.13.13; extra == 'workflow'
+Description-Content-Type: text/markdown
+
+# TrueFoundry
+
+TrueFoundry library to help you interact with the platform programmatically by
+
+- Interacting with the deployments side of TrueFoundry, enabling you to manage workspaces, deployments, applications, and view logs.
+- Providing experiment tracking capabilities, allowing you to track ML experiments and interact with ML repositories within TrueFoundry.
+
+You can access the health of your service, monitoring links, deployed endpoints and track metadata and artifacts by logging on to TrueFoundry's dashboard.
+
+## Installation
+
+To install the `TrueFoundry CLI`, you can use pip:
+
+```bash
+pip install truefoundry
+```
+
+To install the `TrueFoundry CLI` with Workflow deployment features, you can use pip:
+
+```bash
+pip install truefoundry[workflow]
+```
+
+## Examples
+
+https://github.com/truefoundry/getting-started-examples
+
+## Documentation
+
+https://docs.truefoundry.com/

{truefoundry-0.5.7rc1.dist-info → truefoundry-0.5.8rc2.dist-info}/RECORD

@@ -1,14 +1,17 @@
 truefoundry/__init__.py,sha256=Go9M0PvV9kYX8dhldJzpH0RAYB_Pc1kVOxfNlsps89c,321
+truefoundry/logger.py,sha256=u-YCNjg5HBwE70uQcpjIG64Ghos-K2ulTWaxC03BSj4,714
+truefoundry/pydantic_v1.py,sha256=jSuhGtz0Mbk1qYu8jJ1AcnIDK4oxUsdhALc4spqstmM,345
+truefoundry/version.py,sha256=bqiT4Q-VWrTC6P4qfK43mez-Ppf-smWfrl6DcwV7mrw,137
 truefoundry/autodeploy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+truefoundry/autodeploy/cli.py,sha256=9ZxKu_MGIpraMzaW4ZyuQZhlKIQYE3biBrBV4S1h6Fo,14167
+truefoundry/autodeploy/constants.py,sha256=vTh2nA7cjqghqbW2rNh3FbtcIk2scdAWZuuQCmVBO80,1273
+truefoundry/autodeploy/exception.py,sha256=fa_ZyTDUKiMKG2Uayynk1yWcEMsuVluqk2GtV4tfTPU,158
+truefoundry/autodeploy/logger.py,sha256=tkV2UKcOTFl5nz0cn4eRbzxF-2CZd8b7MK9vnhaflYw,325
 truefoundry/autodeploy/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/autodeploy/agents/base.py,sha256=lYw6bKhfCWff5BbtXXn2k4MjMbRIUkOwPBjRT70iXE8,6420
 truefoundry/autodeploy/agents/developer.py,sha256=tO9vmgGjoaqK4rX-DFCyMrViFC2Oqofd89WoXqsEm-A,4142
 truefoundry/autodeploy/agents/project_identifier.py,sha256=KlF7FwrGHDCjF3y3-YaCDxZ8kE34s6XPBagXhG2U1aM,4520
 truefoundry/autodeploy/agents/tester.py,sha256=lqvVBuVcFRvGZnSLAUF2r-1-Yti20o_cbybU19IlSXo,2418
-truefoundry/autodeploy/cli.py,sha256=9ZxKu_MGIpraMzaW4ZyuQZhlKIQYE3biBrBV4S1h6Fo,14167
-truefoundry/autodeploy/constants.py,sha256=vTh2nA7cjqghqbW2rNh3FbtcIk2scdAWZuuQCmVBO80,1273
-truefoundry/autodeploy/exception.py,sha256=fa_ZyTDUKiMKG2Uayynk1yWcEMsuVluqk2GtV4tfTPU,158
-truefoundry/autodeploy/logger.py,sha256=tkV2UKcOTFl5nz0cn4eRbzxF-2CZd8b7MK9vnhaflYw,325
 truefoundry/autodeploy/tools/__init__.py,sha256=9zJiC1d4bv9EL-p5XTCa9fAQ6ZKV--AbgeLz9bBBkyQ,875
 truefoundry/autodeploy/tools/ask.py,sha256=MxUFLP7rjpdJ85gCc3El0wUqTZDjpjAw7WOTdV4LLWE,856
 truefoundry/autodeploy/tools/base.py,sha256=Ltzj2d6DO1gUMx_qSt2c2Zq91MJ5V4FJXpZwl4T3F8M,795
@@ -24,7 +27,7 @@ truefoundry/autodeploy/utils/client.py,sha256=PvbSkfgAjAogGjisinqmh4mP4svowxAC0I
 truefoundry/autodeploy/utils/diff.py,sha256=Ef8Y-VffDKel_-q-GxRam6gqiv8qTLMcqVg6iifXfcA,5358
 truefoundry/autodeploy/utils/pydantic_compat.py,sha256=hEAUy5kLjhPdzw7yGZ2iXGMXbbMVXVlGzIofmyHafXQ,412
 truefoundry/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/cli/__main__.py,sha256=0nEwE0UkFagCO5ctGcqEThdmBT2fN8MVcr0ERItnsNQ,3483
+truefoundry/cli/__main__.py,sha256=oPtY1egJ21nN46me7E-xLfU0m_Zy5Oz2jj571os8Yxg,3465
 truefoundry/cli/config.py,sha256=tf8w4UfVzcC6eYkENvuuCPYt_V3sqVpO1bclORV9tAk,206
 truefoundry/cli/console.py,sha256=9-dMy4YPisCJQziRKTg8Qa0UJnOGl1soiUnJjsnLDvE,242
 truefoundry/cli/const.py,sha256=dVHPo1uAiDSSMXwXoT2mR5kNQjExT98QNVRz98Hz_Ts,510
@@ -34,34 +37,36 @@ truefoundry/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 truefoundry/common/auth_service_client.py,sha256=N3YxKlx63r6cPZqbgb2lqBOPI69ShB7D7RCIq4FSCjc,7949
 truefoundry/common/constants.py,sha256=WQjylPv7xTVYURv4zicJUHYL73Gv3c3i6Iw4oy9Lhg4,2744
 truefoundry/common/credential_file_manager.py,sha256=1yEk1Zm2xS4G0VDFwKSZ4w0VUrcPWQ1nJnoBaz9xyKA,4251
-truefoundry/common/credential_provider.py,sha256=R666i1ADKb2ldwEVh9lnaFcPaa4q3vAscIX3MX1UiFA,4097
-truefoundry/common/entities.py,sha256=8O-EGPk4PKqnyoFMKUTxISCU19rz0KBnfRDJU695DhY,3797
+truefoundry/common/credential_provider.py,sha256=_OhJ2XFlDaVsrUO-FyywxctcGGqDdC2pgcvwEKqQD0Q,4071
+truefoundry/common/entities.py,sha256=12olLKzRT092Jnze5busK68cwH8wwyL6ch77ohKOFPU,3723
 truefoundry/common/exceptions.py,sha256=jkU0N7hV_P-EhXeud4I5vuB9glXXZSWPf8LcH04mSbw,459
 truefoundry/common/request_utils.py,sha256=e9qrAQ1MutU7JALDKcucmNd0KQEVBqgW3yx0w1zeHIU,5700
-truefoundry/common/servicefoundry_client.py,sha256=295ZXOEM46lXZP59O04XFTDCsZLtz3MssXRDMk2x9hg,3169
+truefoundry/common/servicefoundry_client.py,sha256=2fYhdVPSvLXz5C5tosOq86JD8WM3IRUIy1VO9deDxZI,3340
 truefoundry/common/session.py,sha256=BiTJ2WEzLkTvzpqy67xX3fXgiKH_ltzzmQdFG7xUtwI,2904
 truefoundry/common/storage_provider_utils.py,sha256=yURhMw8k0FLFvaviRHDiifhvc6GnuQwGMC9Qd2uM440,10934
-truefoundry/common/utils.py,sha256=sZyg-vpShVXvnrwnINrdsyWJR3FHMUBiOB1CslYdQTo,6072
+truefoundry/common/types.py,sha256=BMJFCsR1lPJAw66IQBSvLyV4I6o_x5oj78gVsUa9si8,188
+truefoundry/common/utils.py,sha256=E1kKw_hqlVMgwyPkER2SvVLIoQ-B_4TQCuM5IJXvgig,6046
 truefoundry/common/warnings.py,sha256=rs6BHwk7imQYedo07iwh3TWEOywAR3Lqhj0AY4khByg,504
 truefoundry/deploy/__init__.py,sha256=e0EgG-fVJ6N9PNIlBINO6QtIMEehQgmz3ONPPjYlcSY,2596
+truefoundry/deploy/python_deploy_codegen.py,sha256=qJHH1BJQII9e6PhkcRFYiE_3De7_VMMm8nM4AX5Eq1o,6513
 truefoundry/deploy/auto_gen/models.py,sha256=83OoBHcu3VP8Xe75vBNaQmPGjqxhsSkHAZZZD4mROOM,87620
 truefoundry/deploy/builder/__init__.py,sha256=1qjHMNBE1poRCZW0WrG46dFM1f1IlivD5352qzsioMU,4953
+truefoundry/deploy/builder/constants.py,sha256=amUkHoHvVKzGv0v_knfiioRuKiJM0V0xW0diERgWiI0,508
+truefoundry/deploy/builder/docker_service.py,sha256=sm7GWeIqyrKaZpxskdLejZlsxcZnM3BTDJr6orvPN4E,3948
+truefoundry/deploy/builder/utils.py,sha256=D68-bqM0NQx-Elg-56mtkENyVyg9faZ9tgTmBuo1Sjs,1076
 truefoundry/deploy/builder/builders/__init__.py,sha256=tlFLXqyDaKLd4iZbo4Hcu_8gOmgtL6drnXpbmQ6x1P8,636
 truefoundry/deploy/builder/builders/dockerfile.py,sha256=AXXTziCkaqIhuM_bwyD1vT1znOwemN1TKgU7eyo-KuM,1522
 truefoundry/deploy/builder/builders/tfy_notebook_buildpack/__init__.py,sha256=x_GwRFKz-Kb4-ZlxOFjBlr0mTgUDe_hVeG4dsIbHo8c,1796
 truefoundry/deploy/builder/builders/tfy_notebook_buildpack/dockerfile_template.py,sha256=rQgdvKmAT9HArVW4TAG5yd2QTKRs3S5LJ9RQbc_EkHE,2518
 truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py,sha256=9r1PYahn-HfzpMth6NkvJoycmmHQpSl0vIVZxWF12xI,1864
 truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=X9mKH7SGZLGZGvY0S2aPB3sgWdSR9TY4eVXtqz_idEA,8973
-truefoundry/deploy/builder/constants.py,sha256=amUkHoHvVKzGv0v_knfiioRuKiJM0V0xW0diERgWiI0,508
-truefoundry/deploy/builder/docker_service.py,sha256=OI8efqK0Gnoii8bcHihpA2StwHVzsMREfBk7NvMR4hY,3950
-truefoundry/deploy/builder/utils.py,sha256=D68-bqM0NQx-Elg-56mtkENyVyg9faZ9tgTmBuo1Sjs,1076
 truefoundry/deploy/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/cli/commands/__init__.py,sha256=wn_TsmrImj9pER16O5aK3aW2mfqbRgrhv-jn4x3_2oM,1278
 truefoundry/deploy/cli/commands/apply_command.py,sha256=Y2e_C8HVpo8CssVod-3JRz-89qStC5JRaNzJ7O2mRlY,2039
 truefoundry/deploy/cli/commands/build_command.py,sha256=zJBywMatbpUlXx5O2aqpEVmPeBIJ9RNnG9abSc8C8CE,1234
 truefoundry/deploy/cli/commands/create_command.py,sha256=rCajvQvAfZU10nDZOYpRACbAATH1zj52ihTWrhnLLUc,1829
 truefoundry/deploy/cli/commands/delete_command.py,sha256=JMibxvt1rsZ8sODa0jI1e6vavjN8Pv04ttVFURSTKTw,2362
-truefoundry/deploy/cli/commands/deploy_command.py,sha256=JW7_gUwM7sg7cNkK-28AdNdgFj26Oj2KjkqIO-CvZFg,4172
+truefoundry/deploy/cli/commands/deploy_command.py,sha256=8aTBvzPaT9xg6KPmpcpqJlmdj4yXzWUfAy6slcoPN74,4123
 truefoundry/deploy/cli/commands/deploy_init_command.py,sha256=g-jBfrEmhZ0TDWsyqPDn4K6q33EqJSGmBTt1eMYig-w,600
 truefoundry/deploy/cli/commands/get_command.py,sha256=HZQGWib-qrS0RdzctRDwDLfMGu0wqWkAM1u26cTZlsc,5944
 truefoundry/deploy/cli/commands/list_command.py,sha256=zKu_JWY35eMIzgSNFYtmwi2uezZ4k-8yk3C1Vqsvshc,4470
@@ -80,36 +85,53 @@ truefoundry/deploy/io/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 truefoundry/deploy/io/output_callback.py,sha256=V2YwUFec4G4a67lM4r-x_64AqdOVNo_9sTdfQWLlvi0,604
 truefoundry/deploy/io/rich_output_callback.py,sha256=TJLiRD-EnFVwgcepxR7WN0koKqW1X2DevETPhNPi_nU,829
 truefoundry/deploy/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=twgEUKSES8HWewMj-8xeD1O3AOz9kfhaaZA4D0BATxk,26444
 truefoundry/deploy/lib/const.py,sha256=Wg0GDnfFu-g1fJr4lU80NH2ULw0R0dYjV7LnW-PbOeM,173
+truefoundry/deploy/lib/logs_utils.py,sha256=SQxRv3jDDmgHdOUMhlMaAPGYskybnBUMpst7QU_i_sc,1469
+truefoundry/deploy/lib/messages.py,sha256=nhp0bCYf_XpUM68hTq5lBY-__vtEyV2uP7NgnJXJ_Vg,925
+truefoundry/deploy/lib/session.py,sha256=FBSq3bxf3my_613NzgtPoeeaAYLGZEoF0n7XccTFNf4,5141
+truefoundry/deploy/lib/util.py,sha256=J7r8San2wKo48A7-BlH2-OKTlBO67zlPjLEhMsL8os0,1059
+truefoundry/deploy/lib/win32.py,sha256=1RcvPTdlOAJ48rt8rCbE2Ufha2ztRqBAE9dueNXArrY,5009
+truefoundry/deploy/lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=xW_JGOk28bVMqdRiggCYMJFIx7iPjOUCdg8DDxfpjmo,26341
 truefoundry/deploy/lib/dao/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/dao/application.py,sha256=oMszpueXPUfTUuN_XdKwoRjQyqAgWHhZ-10cbprCVdM,9226
 truefoundry/deploy/lib/dao/apply.py,sha256=Jadfa3DcbIRz-Bb4Tlixed179HLDnmDXOlrvSxfEgQ0,3021
 truefoundry/deploy/lib/dao/version.py,sha256=AtdW_4O1DPUKdfv2qy6iUJsZ_95vM6z0AqeEy3WDKs8,1130
 truefoundry/deploy/lib/dao/workspace.py,sha256=6YvfCgWDzAULI3Q6JswyZmP1CwJ5rM-ANsIFkbQia0Q,2349
-truefoundry/deploy/lib/logs_utils.py,sha256=SQxRv3jDDmgHdOUMhlMaAPGYskybnBUMpst7QU_i_sc,1469
-truefoundry/deploy/lib/messages.py,sha256=nhp0bCYf_XpUM68hTq5lBY-__vtEyV2uP7NgnJXJ_Vg,925
 truefoundry/deploy/lib/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/model/entity.py,sha256=3YRdyvMYI1aUXUu2tB5b5lZWwpxKdAhjpT79gzTCZGo,8373
-truefoundry/deploy/lib/session.py,sha256=FBSq3bxf3my_613NzgtPoeeaAYLGZEoF0n7XccTFNf4,5141
-truefoundry/deploy/lib/util.py,sha256=J7r8San2wKo48A7-BlH2-OKTlBO67zlPjLEhMsL8os0,1059
-truefoundry/deploy/lib/win32.py,sha256=1RcvPTdlOAJ48rt8rCbE2Ufha2ztRqBAE9dueNXArrY,5009
-truefoundry/deploy/python_deploy_codegen.py,sha256=qJHH1BJQII9e6PhkcRFYiE_3De7_VMMm8nM4AX5Eq1o,6513
 truefoundry/deploy/v2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/v2/lib/__init__.py,sha256=WEiVMZXOVljzEE3tpGJil14liIn_PCDoACJ6b3tZ6sI,188
-truefoundry/deploy/v2/lib/deploy.py,sha256=Uk4osyx7uGfYWgyzd6FlmDEnR6tnIjDdytIlCQAq9Xs,12441
-truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=NM6MDdJ7qeqr1HPM5dLLhwkshQ3OweKQOFjzRZ_2c5w,12797
-truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=6zLJmzRfKXoYa151lc_WUIUOc4wBtNthFfGS9IWMrQc,3990
-truefoundry/deploy/v2/lib/models.py,sha256=pSolLMTArDuYpeNsmeeS5DWliloN_iCDfZSpRllMHUg,1120
-truefoundry/deploy/v2/lib/patched_models.py,sha256=-YnzXSabKs8XDx2I5zYjSKj1lM2qf-L4huewtHSvnP4,16310
-truefoundry/deploy/v2/lib/source.py,sha256=wArU8AwSy7U9EUGLEMmB4Ke7ZwgZWmh8fFU-PACogIk,9347
-truefoundry/logger.py,sha256=u-YCNjg5HBwE70uQcpjIG64Ghos-K2ulTWaxC03BSj4,714
+truefoundry/deploy/v2/lib/deploy.py,sha256=OvotIwCzluaOqX3xceV5sb86WFXYNs4fXn6FJsR9KBU,12420
+truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=92aA8ji46-KjC3tu46KeJuvOFkajSU1WAdLKgqVOGh8,12869
+truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=hqfPxQvuA9rTjaxZY9YqV51hdNFT-0YBXyR-S859PTo,4060
+truefoundry/deploy/v2/lib/models.py,sha256=ogc1UYs1Z2nBdGSKCrde9sk8d0GxFKMkem99uqO5CmM,1148
+truefoundry/deploy/v2/lib/patched_models.py,sha256=v26Rm0BaV6RKr-TxKZ8O3gT-90JA91K4bnGSKgCrQ7o,16338
+truefoundry/deploy/v2/lib/source.py,sha256=zM8k3hJ2N5b8FgkVsYPhEvHPA4rt2rSqtLUMRcaGGiQ,9405
 truefoundry/ml/__init__.py,sha256=ssUEIs8BixPWxynKoeSh-dkRl6AtLXG0PBGYnUR5Az8,2217
+truefoundry/ml/constants.py,sha256=vDq72d4C9FSWqr9MMdjgTF4TuyNFApvo_6RVsSeAjB4,2837
+truefoundry/ml/entities.py,sha256=si5GAqZsWzKu5MPrU4Hk6se7bebHOYhTiNw69ai-Uk8,1485
+truefoundry/ml/enums.py,sha256=arqDkF8duU_oVLFeYMhcfWYbF6Nq5mmjwupJMIheyXM,1790
+truefoundry/ml/exceptions.py,sha256=QpDJSWmF7dIsByS0qOQbQZ_jytdNTzkHDDO3BxhTSo0,332
+truefoundry/ml/git_info.py,sha256=jvAVm9ilqivnGq8qJdUvYdd8Siv0PLtqurB-PXsS5ho,2023
+truefoundry/ml/internal_namespace.py,sha256=QcqMHp6-C2im2H_02hlhi01EIcr1HhNaZprszs13EMU,1790
+truefoundry/ml/logger.py,sha256=VT-BF3BnBYTWVq87O58F0c8uXMu94gYzsiFlGY3_7Ao,458
+truefoundry/ml/mlfoundry_api.py,sha256=PUBGDcjrZsJKQYy1PjT8NYGYxrTvhhHzQ-0jLSl7-zc,61279
+truefoundry/ml/mlfoundry_run.py,sha256=w9_Tph48eTDjna-rZ1fPa6PtX4JzhF-t5PS2hmbK2rY,44319
+truefoundry/ml/model_framework.py,sha256=JGmzdG7o5P6CJr5EYTUOmuly53FfhpoNVqKrhfijAVg,18972
+truefoundry/ml/run_utils.py,sha256=0W208wSLUrbdfk2pjNcZlkUi9bNxG2JORqoe-5rVqHI,2423
+truefoundry/ml/session.py,sha256=eQo1zG5nd0-8Ok68uEBCQDPAWYeJkpzqHak-DQX1ENU,4553
+truefoundry/ml/validation_utils.py,sha256=J5atNhcJLvKj64ralSV9Y5Fv1Rt4SE237ICdP9-7sP4,12149
 truefoundry/ml/artifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=myDpr3RZzEkd_UG9C3rZ4UKEFxXopO5HMVJNpNDiuAo,37737
 truefoundry/ml/autogen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+truefoundry/ml/autogen/client_README.md,sha256=RpZTi__4UauH1kPvz7LfixtrbmTreiVpc2rDyDWstTs,37139
 truefoundry/ml/autogen/client/__init__.py,sha256=Kh1qP7S8L-e7ItofskXQf3y7QFfmxGPFnv5ACAPzwKE,20123
+truefoundry/ml/autogen/client/api_client.py,sha256=M31IycWorn10FYS8WbO_Whaipr0tVu2pVWFCfbS7XCo,29167
+truefoundry/ml/autogen/client/api_response.py,sha256=KRyvecPMXF05PaxILHZ8JHoP4rgKBjKONMgG83aU-rM,844
+truefoundry/ml/autogen/client/configuration.py,sha256=V1oaEnxt-NfpaNmp-EZpf2glovzVhM2coWYt8HBNB4M,15723
+truefoundry/ml/autogen/client/exceptions.py,sha256=XbCbDHhYT3BVejdoGNPgEa4oS56ypkwFdxk1iOc_tFY,5355
+truefoundry/ml/autogen/client/rest.py,sha256=9goba8qHjQuVx5O_yRaTKu7PvBnb7r7swfy3dwuTEgk,14281
 truefoundry/ml/autogen/client/api/__init__.py,sha256=NyMBxBmIzB1o5LzZZwz9LiydHB3-hPqc_sevsnY6Jrw,746
 truefoundry/ml/autogen/client/api/auth_api.py,sha256=zpWzJhUmW6HHMY_atlUf0B25k77E1kue2hmix5I5Ih0,7017
 truefoundry/ml/autogen/client/api/deprecated_api.py,sha256=mu5x_skNcwz8v1Tr6VP72-tVP7pmBV9muyKy_2NH3F0,38824
@@ -120,10 +142,6 @@ truefoundry/ml/autogen/client/api/metrics_api.py,sha256=q3L38eD-2hu4_9YvcSdnWDYX
 truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py,sha256=T8_cYdrd60KzDddRxX3nDeIQe6ihMq-jD8BOAAmzPHo,309716
 truefoundry/ml/autogen/client/api/run_artifacts_api.py,sha256=x-vVnY2LEFChZxiiFauswRWwFz6Qqh30PKXjzuTvxmc,8799
 truefoundry/ml/autogen/client/api/runs_api.py,sha256=-aghrZ2VYuZOw_vBtOzWDsnK7Ji29oZQxK2CLRgyo2w,119232
-truefoundry/ml/autogen/client/api_client.py,sha256=M31IycWorn10FYS8WbO_Whaipr0tVu2pVWFCfbS7XCo,29167
-truefoundry/ml/autogen/client/api_response.py,sha256=KRyvecPMXF05PaxILHZ8JHoP4rgKBjKONMgG83aU-rM,844
-truefoundry/ml/autogen/client/configuration.py,sha256=V1oaEnxt-NfpaNmp-EZpf2glovzVhM2coWYt8HBNB4M,15723
-truefoundry/ml/autogen/client/exceptions.py,sha256=XbCbDHhYT3BVejdoGNPgEa4oS56ypkwFdxk1iOc_tFY,5355
 truefoundry/ml/autogen/client/models/__init__.py,sha256=5LUnBRz-1KBea5KicUDeJ0LPDUKvHJYd6L0iP1q-rSs,18745
 truefoundry/ml/autogen/client/models/add_custom_metrics_to_model_version_request_dto.py,sha256=_ISDspicTGjBCYYXubKfRYYSSQVyW3AvG-jFh47-Zfc,2163
 truefoundry/ml/autogen/client/models/add_features_to_model_version_request_dto.py,sha256=OT1-98DyWNfAHz_EgD2gMX2XkrGQ4Re945fhoAl8qSE,2099
@@ -306,8 +324,6 @@ truefoundry/ml/autogen/client/models/validation_error_loc_inner.py,sha256=nThJ5G
 truefoundry/ml/autogen/client/models/xg_boost_framework.py,sha256=a0EHzncxe_hSFUCAk3nFOMuh8ztQQQBLbTfg0IbwTE8,3324
 truefoundry/ml/autogen/client/models/xg_boost_model_schema.py,sha256=QKJkiERNqF8UiolX2FjvhkUkUlUgDoo3CpTYPAdCxV8,2735
 truefoundry/ml/autogen/client/models/xg_boost_serialization_format.py,sha256=wFFSH0Z9MP8eT_fysHfUu4iqdjEZART-HxTH4e4gNI8,872
-truefoundry/ml/autogen/client/rest.py,sha256=9goba8qHjQuVx5O_yRaTKu7PvBnb7r7swfy3dwuTEgk,14281
-truefoundry/ml/autogen/client_README.md,sha256=RpZTi__4UauH1kPvz7LfixtrbmTreiVpc2rDyDWstTs,37139
 truefoundry/ml/autogen/entities/artifacts.py,sha256=TOm71ZrNo36le3YyMEsE1Cs_BYyMLkFQ7hQtAaMoB9M,21251
 truefoundry/ml/autogen/models/__init__.py,sha256=--TGRea9SQBWFfwtcl3ekb1XGfFTdEkQGSG8a2SJ60I,187
 truefoundry/ml/autogen/models/exceptions.py,sha256=q3n7FGBrg_hUy1UyoefhMnhcXUAaqXlLIGHoOVzn_d8,1390
@@ -316,19 +332,16 @@ truefoundry/ml/autogen/models/signature.py,sha256=rBjpxUIsEeWM0sIyYG5uCJB18DKHR4
 truefoundry/ml/autogen/models/utils.py,sha256=c7RtSLXhOLcP8rjuUtfnMdaKVTZvvbsmw98gPAkAFrs,24371
 truefoundry/ml/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/cli/cli.py,sha256=MwpY7z_NEeJE_XIP7XbZELjNeu2vpMmohttHCKDRk54,335
+truefoundry/ml/cli/utils.py,sha256=j6_mZ4Spn114mz3P4QQ8jx0tmorXIuyQnHXVUSDvZi4,1035
 truefoundry/ml/cli/commands/__init__.py,sha256=diDUiRUX4l6TtNLI4iF-ZblczkELM7FRViJ-8gGNJQY,82
 truefoundry/ml/cli/commands/download.py,sha256=N9MhsEQ3U24v_OmnMZT8Q4SoAi38Sm7a21unrACOSDw,2573
 truefoundry/ml/cli/commands/model_init.py,sha256=9fHCernmQswjEhREzdrnKxwCCWkgmFrpL29H52Sd1gQ,2663
-truefoundry/ml/cli/utils.py,sha256=j6_mZ4Spn114mz3P4QQ8jx0tmorXIuyQnHXVUSDvZi4,1035
 truefoundry/ml/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/clients/servicefoundry_client.py,sha256=Rx-tjiDxNwUgOCDITjtL0vmZcQgJjh_tUdReE23TKIk,2363
-truefoundry/ml/constants.py,sha256=vDq72d4C9FSWqr9MMdjgTF4TuyNFApvo_6RVsSeAjB4,2837
-truefoundry/ml/entities.py,sha256=si5GAqZsWzKu5MPrU4Hk6se7bebHOYhTiNw69ai-Uk8,1485
-truefoundry/ml/enums.py,sha256=arqDkF8duU_oVLFeYMhcfWYbF6Nq5mmjwupJMIheyXM,1790
-truefoundry/ml/exceptions.py,sha256=QpDJSWmF7dIsByS0qOQbQZ_jytdNTzkHDDO3BxhTSo0,332
-truefoundry/ml/git_info.py,sha256=jvAVm9ilqivnGq8qJdUvYdd8Siv0PLtqurB-PXsS5ho,2023
-truefoundry/ml/internal_namespace.py,sha256=QcqMHp6-C2im2H_02hlhi01EIcr1HhNaZprszs13EMU,1790
 truefoundry/ml/log_types/__init__.py,sha256=g4u4D4Jaj0aBK5GtrLV88-qThKZR9pSZ17vFEkN-LmM,125
+truefoundry/ml/log_types/plot.py,sha256=KsVJiZ1uXvMoAumjDTT7EpjWoagtRWxF5rRQDTCRGOY,7571
+truefoundry/ml/log_types/pydantic_base.py,sha256=eBlw_AEyAz4iJKDP4zgJOCFWcldwQqpf7FADW1jzIQY,272
+truefoundry/ml/log_types/utils.py,sha256=xjJ21jdPScvFmw3TbVh5NCzbzJwaqiXJyiiT4xxX1EI,335
 truefoundry/ml/log_types/artifacts/artifact.py,sha256=wIGmuwvf7-sgJ4WJSwhSYk3EBccJAdgcTG6wUiomBKg,22032
 truefoundry/ml/log_types/artifacts/constants.py,sha256=qKxQ5mMvJE4j83BvGW3qNTKunxCiBg_EEjTdgbgJtyE,1036
 truefoundry/ml/log_types/artifacts/dataset.py,sha256=a4dxd2EN8p7Ci-cLGGiDOboN3t0395_XhWE1dmTw1Q4,13112
@@ -340,34 +353,17 @@ truefoundry/ml/log_types/image/constants.py,sha256=wLtGEOA4T5fZHSlOXPuNDLX3lpbCt
 truefoundry/ml/log_types/image/image.py,sha256=qQnAVgErAq4Jn6wXFFpaveOd52zcjUuomUCqNRxO2io,12478
 truefoundry/ml/log_types/image/image_normalizer.py,sha256=vrzfuSpVGgIxw_Q2sbFe7kQ_JpAndX0bMwC7wtfi41g,3104
 truefoundry/ml/log_types/image/types.py,sha256=inFQlyAyDvZtfliFpENirNCm1XO9beyZ8DNn97DoDKs,1568
-truefoundry/ml/log_types/plot.py,sha256=HuYvvRA5r8V0xAIuuqMME2IHb9d3SfGHUiuEkOP3Uks,7515
-truefoundry/ml/log_types/pydantic_base.py,sha256=eBlw_AEyAz4iJKDP4zgJOCFWcldwQqpf7FADW1jzIQY,272
-truefoundry/ml/log_types/utils.py,sha256=xjJ21jdPScvFmw3TbVh5NCzbzJwaqiXJyiiT4xxX1EI,335
-truefoundry/ml/logger.py,sha256=VT-BF3BnBYTWVq87O58F0c8uXMu94gYzsiFlGY3_7Ao,458
-truefoundry/ml/mlfoundry_api.py,sha256=PUBGDcjrZsJKQYy1PjT8NYGYxrTvhhHzQ-0jLSl7-zc,61279
-truefoundry/ml/mlfoundry_run.py,sha256=M-0FkZYDxwv5EZUDtVqiV8JxC_3BWR80N7Kp4Yj7bPY,44291
-truefoundry/ml/model_framework.py,sha256=JGmzdG7o5P6CJr5EYTUOmuly53FfhpoNVqKrhfijAVg,18972
-truefoundry/ml/run_utils.py,sha256=0W208wSLUrbdfk2pjNcZlkUi9bNxG2JORqoe-5rVqHI,2423
-truefoundry/ml/session.py,sha256=eQo1zG5nd0-8Ok68uEBCQDPAWYeJkpzqHak-DQX1ENU,4553
-truefoundry/ml/validation_utils.py,sha256=J5atNhcJLvKj64ralSV9Y5Fv1Rt4SE237ICdP9-7sP4,12149
-truefoundry/pydantic_v1.py,sha256=jSuhGtz0Mbk1qYu8jJ1AcnIDK4oxUsdhALc4spqstmM,345
-truefoundry/version.py,sha256=bqiT4Q-VWrTC6P4qfK43mez-Ppf-smWfrl6DcwV7mrw,137
 truefoundry/workflow/__init__.py,sha256=XY83vqtLAclI82atZXyBtF9ZgLROXaaXO5p60XH5hJA,1493
 truefoundry/workflow/container_task.py,sha256=8arieePsX4__OnG337hOtCiNgJwtKJJCsZcmFmCBJtk,402
-truefoundry/workflow/example/deploy.sh,sha256=wfbPRrCi04WYRqCf4g-Xo12uWbcqPD6G_Tz0lV0jU_U,60
-truefoundry/workflow/example/hello_world_package/workflow.py,sha256=IkRKfPY5BcvLPo_PVuNbZKK9PPJ93LRkzb1a3RKQYOw,435
-truefoundry/workflow/example/package/test_workflow.py,sha256=TUcvYCGz4IbzF9bQJAfB-6Wd2wx-0RKcuIpUG0t6d8c,3972
-truefoundry/workflow/example/truefoundry.yaml,sha256=LlPrMADSPJsiXRoK76N_RVjX1bnZ3FH1u2jXrwLfR9I,226
-truefoundry/workflow/example/workflow.yaml,sha256=YtYdKXMuW_08gfEo21XSculj2MGI2lfEnGF8qCT8NKE,2858
 truefoundry/workflow/map_task.py,sha256=2m3qGXQ90k9LdS45q8dqCCECc3qr8t2m_LMCVd1mZ7g,1737
 truefoundry/workflow/python_task.py,sha256=SRXRLC4vdBqGjhkwuaY39LEWN6iPCpJAuW17URRdWTY,1128
+truefoundry/workflow/task.py,sha256=34m55mALXx6ko9o5HkK6FDtMajdvJzBhOsHwDM2RcBA,1779
+truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
 truefoundry/workflow/remote_filesystem/__init__.py,sha256=LQ95ViEjJ7Ts4JcCGOxMPs7NZmQdZ4bTiq6qXtsjUhE,206
 truefoundry/workflow/remote_filesystem/logger.py,sha256=em2l7D6sw7xTLDP0kQSLpgfRRCLpN14Qw85TN7ujQcE,1022
 truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=xcT0wQmQlgzcj0nP3tJopyFSVWT1uv3nhiTIuwfXYeg,12342
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=nSGPZu0Gyd_jz0KsEE-7w_BmnTD8CVF1S8cUJoxaCbc,13305
-truefoundry/workflow/task.py,sha256=34m55mALXx6ko9o5HkK6FDtMajdvJzBhOsHwDM2RcBA,1779
-truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.5.7rc1.dist-info/METADATA,sha256=x5GilnYFCgdiFnR85bO_VZSTstYfnHbumizGDELekAM,2887
-truefoundry-0.5.7rc1.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-truefoundry-0.5.7rc1.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.5.7rc1.dist-info/RECORD,,
+truefoundry-0.5.8rc2.dist-info/METADATA,sha256=6TAdw_xKTV9A3DU7jZMAJp5yCSW6NyGzDtit4D8_1tg,2523
+truefoundry-0.5.8rc2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.5.8rc2.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.5.8rc2.dist-info/RECORD,,

{truefoundry-0.5.7rc1.dist-info → truefoundry-0.5.8rc2.dist-info}/WHEEL

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

truefoundry-0.5.8rc2.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+tfy = truefoundry.cli.__main__:main
+truefoundry = truefoundry.cli.__main__:main

truefoundry/workflow/example/deploy.sh

@@ -1 +0,0 @@
- tfy deploy --workspace_fqn tfy-ctl-euwe1-devtest:nikhil-ws

truefoundry/workflow/example/hello_world_package/workflow.py

@@ -1,20 +0,0 @@
-from truefoundry.workflow import PythonTaskConfig, TaskPythonBuild, task, workflow
-
-
-@task(
-    task_config=PythonTaskConfig(
-        image=TaskPythonBuild(
-            python_version="3.9",
-            pip_packages=["truefoundry[workflow]"],
-        ),
-        service_account="tfy-workflows-sa",
-    )
-)
-def say_hello() -> str:
-    return "Hello, World!"
-
-
-@workflow
-def hello_world_wf() -> str:
-    res = say_hello()
-    return res

truefoundry/workflow/example/package/test_workflow.py

@@ -1,151 +0,0 @@
-import os
-from functools import partial
-from pathlib import Path
-from typing import List, Optional, Tuple
-
-from truefoundry.deploy import Image, NvidiaGPU, Resources
-from truefoundry.workflow import (
-    ContainerTask,
-    ContainerTaskConfig,
-    FlyteDirectory,
-    PythonTaskConfig,
-    TaskPythonBuild,
-    conditional,
-    map_task,
-    task,
-    workflow,
-)
-
-cpu_task_config = PythonTaskConfig(
-    image=TaskPythonBuild(
-        python_version="3.9",
-        pip_packages=["truefoundry[workflow]==0.4.1"],
-    ),
-    resources=Resources(cpu_request=0.45),
-    service_account="tfy-workflows-sa",
-)
-
-
-# figure out naming
-@task(task_config=cpu_task_config)
-def should_train_tokenizer(tokenizer: str) -> bool:
-    print("Should train tokenizer")
-    return not bool(tokenizer)
-
-
-@task(
-    task_config=cpu_task_config,
-    # currently we will not support the caching, but keeping here for now
-    # cache=True,
-    # cache_version="1.0",
-)
-def train_tokenizer() -> str:
-    print("Training tokenizer")
-    return "trained_tokenizer"
-
-
-@task(
-    task_config=PythonTaskConfig(
-        image=TaskPythonBuild(
-            python_version="3.9",
-            pip_packages=["truefoundry[workflow]==0.4.1", "pynvml==11.5.0"],
-            cuda_version="11.5-cudnn8",
-        ),
-        env={
-            "NVIDIA_DRIVER_CAPABILITIES": "compute,utility",
-            "NVIDIA_VISIBLE_DEVICES": "all",
-        },
-        resources=Resources(cpu_request=0.45, devices=[NvidiaGPU(name="T4", count=1)]),
-        service_account="tfy-workflows-sa",
-    ),
-)
-def train_model(tokenizer: str) -> Tuple[FlyteDirectory, str]:
-    print("Training model")
-    import flytekit
-    # from pynvml import nvmlDeviceGetCount, nvmlInit
-
-    # nvmlInit()
-    # assert nvmlDeviceGetCount() > 0
-
-    working_dir = flytekit.current_context().working_directory
-    local_dir = Path(os.path.join(working_dir, "csv_files"))
-    local_dir.mkdir(exist_ok=True)
-
-    with open(os.path.join(local_dir, "model"), "w", encoding="utf-8") as f:
-        f.write(tokenizer)
-
-    return FlyteDirectory(path=str(local_dir)), "hello"
-
-
-@task(task_config=cpu_task_config)
-def get_validation_data() -> List[str]:
-    print("Getting validation data")
-    return ["foo", "bar", "baz"]
-
-
-@task(task_config=cpu_task_config)
-def validate_model(model: FlyteDirectory, tokenizer: str, validation_data: str) -> bool:
-    print(validation_data)
-    model_path = os.path.join(model, "model")
-    with open(model_path, "r", encoding="utf-8") as f:
-        return f.read() == tokenizer
-
-
-@task(task_config=cpu_task_config)
-def all_good(validations: List[bool]) -> bool:
-    print("Validations", validations)
-    return all(validations)
-
-
-echo = ContainerTask(
-    name="echo",
-    task_config=ContainerTaskConfig(
-        image=Image(
-            image_uri="docker.io/bash:4.1",
-            command=["echo", "hello"],
-        ),
-        service_account="tfy-workflows-sa",
-    ),
-)
-
-
-@task(task_config=cpu_task_config)
-def random(tokenizer: str) -> Tuple[FlyteDirectory, str]:
-    print(tokenizer)
-    return FlyteDirectory(path=""), "random"
-
-
-@workflow(
-    # execution_configs=[
-    #     ExecutionConfig(
-    #         schedule="*/10 * * * *",
-    #     )
-    # ]
-)
-def train(
-    tokenizer: str = "",
-    test_v2: str = "",
-    optional_var: Optional[str] = "",
-    default_v: Optional[str] = "hello1",
-    default_v2: str = "hello2",
-) -> bool:
-    stt = should_train_tokenizer(tokenizer=tokenizer)
-    model, t = (
-        conditional("train_tokenizer")
-        .if_(stt.is_true())
-        .then(train_model(tokenizer=tokenizer))
-        .else_()
-        .then(random(tokenizer=tokenizer))
-    )
-    validation_task = partial(validate_model, model=model, tokenizer=t)
-    validation_data = get_validation_data()
-    validations = map_task(
-        validation_task,
-        concurrency=2,
-    )(validation_data=validation_data)
-    echo()
-    return all_good(validations=validations)
-
-
-if __name__ == "__main__":
-    train()

truefoundry/workflow/example/truefoundry.yaml

@@ -1,9 +0,0 @@
-type: workflow
-name: my-test-workflow9
-source:
-  type: local
-  local_build: false
-  project_root_path: "./"
-workflow_file_path: package/test_workflow.py
-# workflow_file_path: hello_world_package/workflow.py
-flyte_entities: []

truefoundry/workflow/example/workflow.yaml

@@ -1,116 +0,0 @@
-type: workflow
-name: my-test-workflow5
-source:
-  type: remote
-  remote_uri: s3://tfy-mlfoundry-tfy-ctl-euwe1-devtest20221012173221749100000002/tfy-ctl-euwe1-devtest:nikhil-ws/my-test-workflow5/clyhb79w4013l01oyfq1iawd5/package.tar.gz
-workflow_file_path: hello_world_package/workflow.py
-docker_registry: docker.io/nikp-test
-flyte_entities:
-- template:
-    id:
-      resourceType: TASK
-      name: hello_world_package.workflow.say_hello
-    type: python-task
-    metadata:
-      runtime:
-        type: FLYTE_SDK
-        version: 1.12.2
-        flavor: python
-      retries: {}
-    interface:
-      inputs: {}
-      outputs:
-        variables:
-          o0:
-            type:
-              simple: STRING
-            description: o0
-    custom:
-      truefoundry:
-        type: python-task-config
-        image:
-          type: task-python-build
-          python_version: '3.9'
-          pip_packages:
-          - flytekit==1.10.3
-        resources:
-          cpu_request: 0.2
-          cpu_limit: 0.5
-          memory_request: 200
-          memory_limit: 500
-          ephemeral_storage_request: 1000
-          ephemeral_storage_limit: 2000
-    container:
-      image: ':'
-      args:
-      - pyflyte-execute
-      - --inputs
-      - '{{.input}}'
-      - --output-prefix
-      - '{{.outputPrefix}}'
-      - --raw-output-data-prefix
-      - '{{.rawOutputDataPrefix}}'
-      - --checkpoint-path
-      - '{{.checkpointOutputPrefix}}'
-      - --prev-checkpoint
-      - '{{.prevCheckpointPrefix}}'
-      - --resolver
-      - flytekit.core.python_auto_container.default_task_resolver
-      - --
-      - task-module
-      - hello_world_package.workflow
-      - task-name
-      - say_hello
-      resources: {}
-  description:
-    longDescription:
-      format: DESCRIPTION_FORMAT_RST
-- template:
-    id:
-      resourceType: WORKFLOW
-      name: hello_world_package.workflow.hello_world_wf
-    metadata: {}
-    interface:
-      inputs: {}
-      outputs:
-        variables:
-          o0:
-            type:
-              simple: STRING
-            description: o0
-    nodes:
-    - id: n0
-      metadata:
-        name: say_hello
-        retries: {}
-      taskNode:
-        referenceId:
-          resourceType: TASK
-          name: hello_world_package.workflow.say_hello
-        overrides: {}
-    outputs:
-    - var: o0
-      binding:
-        promise:
-          nodeId: n0
-          var: o0
-    metadataDefaults: {}
-  description:
-    longDescription:
-      format: DESCRIPTION_FORMAT_RST
-- id:
-    resourceType: LAUNCH_PLAN
-    name: hello_world_package.workflow.hello_world_wf
-  spec:
-    workflowId:
-      resourceType: WORKFLOW
-      name: hello_world_package.workflow.hello_world_wf
-    entityMetadata: {}
-    defaultInputs: {}
-    fixedInputs: {}
-    labels: {}
-    annotations: {}
-    rawOutputDataConfig: {}
-  closure:
-    expectedInputs: {}
-    expectedOutputs: {}

truefoundry-0.5.7rc1.dist-info/METADATA

@@ -1,79 +0,0 @@
-Metadata-Version: 2.3
-Name: truefoundry
-Version: 0.5.7rc1
-Summary: TrueFoundry CLI
-Author: Abhishek Choudhary
-Author-email: abhishek@truefoundry.com
-Requires-Python: >=3.8.1,<3.13
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Provides-Extra: workflow
-Requires-Dist: GitPython (>=3.1.43,<4.0.0)
-Requires-Dist: Mako (>=1.1.6,<2.0.0)
-Requires-Dist: PyJWT (>=2.0.0,<3.0.0)
-Requires-Dist: PyYAML (>=6.0.0,<7.0.0)
-Requires-Dist: aenum (>=3.0.0,<4.0.0)
-Requires-Dist: click (>=7.0.0,<9.0.0)
-Requires-Dist: coolname (>=1.1.0,<2.0.0)
-Requires-Dist: docker (>=6.1.2,<8.0.0)
-Requires-Dist: filelock (>=3.8.0,<4.0.0)
-Requires-Dist: flytekit (==1.13.13) ; extra == "workflow"
-Requires-Dist: gitignorefile (>=1.1.2,<2.0.0)
-Requires-Dist: importlib-metadata (>=4.11.3,<9.0.0)
-Requires-Dist: importlib-resources (>=5.2.0,<7.0.0)
-Requires-Dist: numpy (>=1.23.0,<2.0.0) ; python_version < "3.12"
-Requires-Dist: numpy (>=1.26.0,<2.0.0) ; python_version >= "3.12"
-Requires-Dist: openai (>=1.16.2,<2.0.0)
-Requires-Dist: packaging (>=20.0,<25.0)
-Requires-Dist: pydantic (>=1.8.2,<3.0.0)
-Requires-Dist: pygments (>=2.12.0,<3.0.0)
-Requires-Dist: python-dateutil (>=2.8.2,<3.0.0)
-Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
-Requires-Dist: python-socketio[client] (>=5.5.2,<6.0.0)
-Requires-Dist: questionary (>=1.10.0,<2.0.0)
-Requires-Dist: requests (>=2.18.0,<3.0.0)
-Requires-Dist: requirements-parser (>=0.11.0,<0.12.0)
-Requires-Dist: rich (>=13.7.1,<14.0.0)
-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 (>=4.0)
-Requires-Dist: urllib3 (>=1.26.18,<3)
-Requires-Dist: yq (>=3.1.0,<4.0.0)
-Description-Content-Type: text/markdown
-
-# TrueFoundry
-
-TrueFoundry library to help you interact with the platform programmatically by
-
-- Interacting with the deployments side of TrueFoundry, enabling you to manage workspaces, deployments, applications, and view logs.
-- Providing experiment tracking capabilities, allowing you to track ML experiments and interact with ML repositories within TrueFoundry.
-
-You can access the health of your service, monitoring links, deployed endpoints and track metadata and artifacts by logging on to TrueFoundry's dashboard.
-
-## Installation
-
-To install the `TrueFoundry CLI`, you can use pip:
-
-```bash
-pip install truefoundry
-```
-
-To install the `TrueFoundry CLI` with Workflow deployment features, you can use pip:
-
-```bash
-pip install truefoundry[workflow]
-```
-
-## Examples
-
-https://github.com/truefoundry/getting-started-examples
-
-## Documentation
-
-https://docs.truefoundry.com/
-

truefoundry-0.5.7rc1.dist-info/entry_points.txt

@@ -1,4 +0,0 @@
-[console_scripts]
-tfy=truefoundry.cli.__main__:main
-truefoundry=truefoundry.cli.__main__:main
-