truefoundry 0.4.3rc0__py3-none-any.whl → 0.4.4__py3-none-any.whl

truefoundry/ml/log_types/artifacts/utils.py

@@ -2,8 +2,6 @@ import json
 import logging
 import os
 import posixpath
-import shutil
-import tempfile
 from pathlib import Path
 from typing import Any, Dict, Optional, Sequence, Tuple, Union
 
@@ -13,20 +11,22 @@ from truefoundry.ml.log_types.artifacts.constants import DESCRIPTION_MAX_LENGTH
 logger = logging.getLogger("truefoundry.ml")
 
 
-def _copy_tree(src_path, dest_path, symlinks=False, ignore_dangling_symlinks=False):
+def _copy_tree(
+    root_dir: str, src_path: str, dest_path: str, dest_to_src: Dict[str, str]
+):
     os.makedirs(dest_path, exist_ok=True)
     for item in os.listdir(src_path):
         src = os.path.join(src_path, item)
         dest = os.path.join(dest_path, item)
         if os.path.isdir(src):
             _copy_tree(
-                src,
-                dest,
-                symlinks=symlinks,
-                ignore_dangling_symlinks=ignore_dangling_symlinks,
+                root_dir=root_dir,
+                src_path=src,
+                dest_path=dest,
+                dest_to_src=dest_to_src,
             )
         else:
-            shutil.copy2(src, dest, follow_symlinks=True)
+            dest_to_src[dest] = src
 
 
 def is_destination_path_dirlike(dest_path) -> bool:
@@ -48,9 +48,9 @@ def _copy_additional_files(
     model_dir: Optional[str],  # relative to files_dir e.g "model/"
     additional_files: Sequence[Tuple[Union[str, Path], Optional[str]]],
     ignore_model_dir_dest_conflict: bool = False,
-):
+    existing_dest_to_src_map: Optional[Dict[str, str]] = None,
+) -> Dict[str, str]:
     """
-
     File copying examples:
         # non ambiguous
         # a.txt -> /tmp/                            result /tmp/a.txt
@@ -69,6 +69,7 @@ def _copy_additional_files(
         # .gitignore -> /tmp/.gitinclude             result /tmp/.gitinclude
         # a.txt -> /tmp/a                            result /tmp/a
     """
+    dest_to_src = existing_dest_to_src_map or {}
     for src_path, dest_path in additional_files:
         src_path = str(src_path)
         if not os.path.exists(src_path):
@@ -99,26 +100,47 @@ def _copy_additional_files(
             _src = src_path
             if is_destination_path_dirlike(dest_abs_path):
                 os.makedirs(dest_abs_path, exist_ok=True)
-                _dst = os.path.relpath(
-                    os.path.join(dest_abs_path, os.path.basename(_src)), files_abs_dir
-                )
+                dest_abs_path = os.path.join(dest_abs_path, os.path.basename(_src))
             else:
                 os.makedirs(os.path.dirname(dest_abs_path), exist_ok=True)
-                _dst = os.path.relpath(dest_abs_path, files_abs_dir)
+            _dst = os.path.relpath(dest_abs_path, files_abs_dir)
             logger.info(f"Adding file {_src} as /{_dst}")
-            shutil.copy2(src_path, dest_abs_path, follow_symlinks=True)
+            dest_to_src[dest_abs_path] = src_path
         elif os.path.isdir(src_path):
             os.makedirs(dest_abs_path, exist_ok=True)
             _src = src_path.rstrip("/")
             _dst = os.path.relpath(dest_abs_path, files_abs_dir).rstrip("/")
             logger.info(f"Adding contents of {_src}/ to /{_dst}/")
             _copy_tree(
+                root_dir=root_dir,
                 src_path=src_path,
                 dest_path=dest_abs_path,
-                symlinks=True,
-                ignore_dangling_symlinks=False,
+                dest_to_src=dest_to_src,
             )
 
+    return dest_to_src
+
+
+def _make_dest_to_src_map_from_dir(root_dir: str) -> Dict[str, str]:
+    dest_to_src_map = {}
+    for root, _, files in os.walk(root_dir):
+        for file in files:
+            src = os.path.join(root, file)
+            dest = src
+            dest_to_src_map[dest] = src
+    return dest_to_src_map
+
+
+def _get_src_dest_pairs(
+    root_dir: str,
+    dest_to_src_map: Dict[str, str],
+) -> Sequence[Tuple[str, str]]:
+    src_dest_pairs = [
+        (src_path, os.path.relpath(dest_abs_path, root_dir))
+        for dest_abs_path, src_path in dest_to_src_map.items()
+    ]
+    return src_dest_pairs
+
 
 def _validate_description(description: Optional[str]):
     if description is not None:
@@ -141,8 +163,8 @@ def _validate_artifact_metadata(metadata: Dict[str, Any]):
         raise MlFoundryException("`metadata` must be json serializable dict") from ve
 
 
-def calculate_local_directory_size(
-    directory: tempfile.TemporaryDirectory,  # type: ignore[type-arg]
+def calculate_total_size(
+    paths: Sequence[str],
 ):
     """
     Tells about the size of the artifact
@@ -153,9 +175,4 @@ def calculate_local_directory_size(
     Returns:
         total size of the artifact
     """
-    total_size = 0
-    for path, _dirs, files in os.walk(directory.name):
-        for f in files:
-            file_path = os.path.join(path, f)
-            total_size += os.stat(file_path).st_size
-    return total_size
+    return sum(os.stat(os.path.realpath(file_path)).st_size for file_path in paths)

truefoundry/ml/log_types/image/image.py

@@ -18,6 +18,7 @@ from truefoundry.ml.log_types.artifacts.constants import (
     FILES_DIR,
     INTERNAL_METADATA_PATH,
 )
+from truefoundry.ml.log_types.artifacts.utils import _make_dest_to_src_map_from_dir
 from truefoundry.ml.log_types.image.constants import (
     DEFAULT_IMAGE_FORMAT,
     IMAGE_KEY_REGEX,
@@ -344,6 +345,7 @@ class Image:
             name=key,
             artifact_type=ArtifactType.IMAGE,
             artifact_dir=temp_dir,
+            dest_to_src_map=_make_dest_to_src_map_from_dir(root_dir=temp_dir.name),
             internal_metadata=internal_metadata,
             step=step,
         )

truefoundry/ml/log_types/plot.py

@@ -18,6 +18,7 @@ from truefoundry.ml.log_types.artifacts.constants import (
     FILES_DIR,
     INTERNAL_METADATA_PATH,
 )
+from truefoundry.ml.log_types.artifacts.utils import _make_dest_to_src_map_from_dir
 from truefoundry.ml.log_types.pydantic_base import PydanticBase
 from truefoundry.ml.log_types.utils import validate_key_name
 from truefoundry.pydantic_v1 import BaseModel
@@ -186,6 +187,7 @@ class Plot:
             name=key,
             artifact_type=ArtifactType.PLOT,
             artifact_dir=temp_dir,
+            dest_to_src_map=_make_dest_to_src_map_from_dir(root_dir=temp_dir.name),
             internal_metadata=internal_metadata,
             step=step,
         )

truefoundry/ml/mlfoundry_api.py

@@ -8,7 +8,7 @@ import coolname
 import pandas as pd
 
 from truefoundry.common.utils import relogin_error_message
-from truefoundry.ml import constants, env_vars
+from truefoundry.ml import constants
 from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
     ArtifactDto,
     ArtifactType,
@@ -59,10 +59,15 @@ from truefoundry.ml.validation_utils import (
 
 _SEARCH_MAX_RESULTS_DEFAULT = 1000
 
+_INTERNAL_ENV_VARS = [
+    "TFY_INTERNAL_APPLICATION_ID",
+    "TFY_INTERNAL_JOB_RUN_NAME",
+]
+
 
 def _get_internal_env_vars_values() -> Dict[str, str]:
     env = {}
-    for env_var_name in env_vars.INTERNAL_ENV_VARS:
+    for env_var_name in _INTERNAL_ENV_VARS:
         value = os.getenv(env_var_name)
         if value:
             env[env_var_name] = value
@@ -330,7 +335,6 @@ class MlFoundry:
         tags.update(_get_internal_env_vars_values())
         _run = self._runs_api.create_run_post(
             CreateRunRequestDto(
-                user_id="unknown",  # This does not matter, because on server we use the id from token
                 start_time=int(
                     time.time() * 1000
                 ),  # TODO (chiragjn): computing start time should be on server side!

truefoundry/ml/session.py

@@ -113,6 +113,8 @@ def _get_api_client(
     session: Optional[Session] = None,
     allow_anonymous: bool = False,
 ) -> ApiClient:
+    from truefoundry.version import __version__
+
     session = session or get_active_session()
     if session is None:
         if allow_anonymous:
@@ -131,7 +133,7 @@ def _get_api_client(
     )
     configuration.retries = urllib3_retry(retries=2)
     api_client = ApiClient(configuration=configuration)
-    api_client.user_agent = "truefoundry-cli"
+    api_client.user_agent = f"truefoundry-cli/{__version__}"
     return api_client
 
 

truefoundry/workflow/__init__.py

@@ -1,4 +1,5 @@
 try:
+    import fsspec
     from flytekit import task as _
 except ImportError:
     print("To use workflows, please run 'pip install truefoundry[workflow]'.")
@@ -6,6 +7,8 @@ except ImportError:
 from flytekit import conditional
 from flytekit.types.directory import FlyteDirectory
 
+from truefoundry.common.constants import ENV_VARS
+from truefoundry.common.tfy_signed_url_fs import SignedURLFileSystem
 from truefoundry.deploy.v2.lib.patched_models import (
     ContainerTaskConfig,
     PythonTaskConfig,
@@ -32,3 +35,10 @@ __all__ = [
     "PythonTaskConfig",
     "ExecutionConfig",
 ]
+
+
+# Register the SignedURLFileSystem implementation for fsspec
+if ENV_VARS.TFY_INTERNAL_SIGNED_URL_SERVER_HOST:
+    fsspec.register_implementation("s3", SignedURLFileSystem, clobber=True)
+    fsspec.register_implementation("gs", SignedURLFileSystem, clobber=True)
+    fsspec.register_implementation("abfs", SignedURLFileSystem, clobber=True)

{truefoundry-0.4.3rc0.dist-info → truefoundry-0.4.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.3rc0
+Version: 0.4.4
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com
@@ -21,7 +21,7 @@ Requires-Dist: coolname (>=1.1.0,<2.0.0)
 Requires-Dist: docker (>=6.1.2,<8.0.0)
 Requires-Dist: fastapi (>=0.56.0,<0.200.0)
 Requires-Dist: filelock (>=3.8.0,<4.0.0)
-Requires-Dist: flytekit (==1.13.8) ; extra == "workflow"
+Requires-Dist: flytekit (==1.12.2) ; 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)

{truefoundry-0.4.3rc0.dist-info → truefoundry-0.4.4.dist-info}/RECORD

@@ -27,23 +27,25 @@ truefoundry/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/cli/__main__.py,sha256=-NkhYlT3mC5MhtekueKAvCw-sWvguj0LJRpXWzvvFjc,727
 truefoundry/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/common/auth_service_client.py,sha256=tZOa0NdATnItsMeTnEnUeTZQIgUJtpU-nvLdWtB4Px8,7978
-truefoundry/common/constants.py,sha256=gnswB-aK5atSRY-GE1Cji6ZCEs2FoK9n1lCX5JuQGqI,484
+truefoundry/common/constants.py,sha256=y8T5S-CoFKubGbSrHXq0VCgr2O3RiEbNV5KP4NDNqhk,2089
 truefoundry/common/credential_file_manager.py,sha256=1yEk1Zm2xS4G0VDFwKSZ4w0VUrcPWQ1nJnoBaz9xyKA,4251
-truefoundry/common/credential_provider.py,sha256=YQ6HKl8ZZFTg48vBZMauEAnM6IrEO3oOzM2DA47N-P0,4071
+truefoundry/common/credential_provider.py,sha256=Aht7hFLsnyRgMR34dRbzln7dor0WYSeA8ej8ApNmnKM,4148
 truefoundry/common/entities.py,sha256=8O-EGPk4PKqnyoFMKUTxISCU19rz0KBnfRDJU695DhY,3797
 truefoundry/common/exceptions.py,sha256=ePpiQ_zmWe4e94gOgeMiyP_AZnKwjEBfyXsB5ScGYcI,329
-truefoundry/common/request_utils.py,sha256=-ss8033PClJOMJuS5Ue1zCFRVK7ZW9vjtva1b5G9uok,2849
-truefoundry/common/servicefoundry_client.py,sha256=klohrELTFTJX-jvl9LGtdIoJiZMZRuQQB4Ho7PJ7u3s,3091
-truefoundry/common/utils.py,sha256=lFiDEVoQDJoZPEN4Zun4GQz2TsfdbXfNHI41BErETq8,3320
+truefoundry/common/request_utils.py,sha256=5xw4YGUcMf71Ncal3OfFCa-PoWDIvG3hYGCDa4Da4OI,2854
+truefoundry/common/servicefoundry_client.py,sha256=2fxmgCM-ckFHpnm6n_mL-5Z8RWN_q-dYVvFC29bkYSg,3120
+truefoundry/common/tfy_signed_url_client.py,sha256=wa9pfZIUquXFE9EdHDOJXjqqU1uJr1PkvI_7vnOjZ9M,9904
+truefoundry/common/tfy_signed_url_fs.py,sha256=Vc7Il077nZIGbDBQ8aTPNtJVcypYaoZHr13vV00REko,8486
+truefoundry/common/utils.py,sha256=vacYbVTEvcEY4niH4C9Wb_aCeaGgh2un3_ApAt556VQ,3621
 truefoundry/deploy/__init__.py,sha256=ugawKF2G02EmEXX35oZ2tec12d9oWN28Sf6mtGGIERY,2281
-truefoundry/deploy/auto_gen/models.py,sha256=AjbjNFwWDTIOjN_E1fSUgP-ej9nsb2OpZPhsnIRAVGs,80621
+truefoundry/deploy/auto_gen/models.py,sha256=4MaxkG2_5Wg6avaZRlK0D4JiVEM5rk3NU0BCiTx8VyU,82477
 truefoundry/deploy/builder/__init__.py,sha256=1qjHMNBE1poRCZW0WrG46dFM1f1IlivD5352qzsioMU,4953
 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=UmMcTY-8MrLY3H5owpn6ax-VePQl4MiMTmHlQ9qEtQw,1742
-truefoundry/deploy/builder/builders/tfy_notebook_buildpack/dockerfile_template.py,sha256=IZbXsSj86SxWRZNQXepEnsOL_eYwWaOq5oh-WiyvL2c,2485
+truefoundry/deploy/builder/builders/tfy_notebook_buildpack/dockerfile_template.py,sha256=rQgdvKmAT9HArVW4TAG5yd2QTKRs3S5LJ9RQbc_EkHE,2518
 truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py,sha256=XNJ3MKWqyWIbFNnUQMpB8oVC5Pt5Wsm_bRKbvkXRIG8,1696
-truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=cJlQK_kgNELULcp2Q_kMWyPNmUuZdhOELDnI0rMiqRc,7473
+truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=nwTpeIrDOZC8_m_VcbQ5Vp_rEO2ioC8FtzOEo17JY7A,7373
 truefoundry/deploy/builder/constants.py,sha256=eIukBjD6I4KvEmAPpdbPlPPr76yhS-uNr3RVFkzEdgs,257
 truefoundry/deploy/builder/docker_service.py,sha256=OI8efqK0Gnoii8bcHihpA2StwHVzsMREfBk7NvMR4hY,3950
 truefoundry/deploy/builder/utils.py,sha256=9RZnkhoHFTRUt_x3nck0aVz7cLpzA3jiwQH-ZZZrjf8,938
@@ -92,7 +94,7 @@ truefoundry/deploy/json_util.py,sha256=x_-7YYQ4_HUIJ8ofOcclAp9JWhgTWjR9Th6Q0FuRq
 truefoundry/deploy/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/auth/servicefoundry_session.py,sha256=5TCYPunAygtn5mb0mp_VcWKEalKMKPbyWMWer-Vty2g,1916
 truefoundry/deploy/lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=vFrd5yeStjT_uWL3JxHrx0iAmNpuv26K7IEnsF1RMM8,26101
+truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=tkuPNbo9vDTgXPHD4p3d-t6HigBvKNfML1pTQXBwJfY,26170
 truefoundry/deploy/lib/const.py,sha256=repGJLuoMqtzeq5tCjjkN4bH187FVHVKI30BricOlvc,244
 truefoundry/deploy/lib/dao/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/dao/application.py,sha256=uUTFSQkLUrFCtQQgS2Jm9BpyHyhMkN4GI1yx9oJo4_E,9161
@@ -103,15 +105,15 @@ truefoundry/deploy/lib/exceptions.py,sha256=ZT2o3mar3BYtnjKHn2wf4rBGEaFIgf-zkQmz
 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=8J8yd98iWtSy8giShdDRNyzbN1UgSXx4XtmZLljdWnE,8552
+truefoundry/deploy/lib/model/entity.py,sha256=fq8hvdJQgQn4uZqxpKrzmaoJhQG53_EbDoDxUPPmOZ0,8387
 truefoundry/deploy/lib/session.py,sha256=Vg6rCA315T0yS0xG4ayJ84Ia_9ZfibH8utOSwPBMAmw,4953
 truefoundry/deploy/lib/util.py,sha256=3TapV7yczkheC1MMMfmJDGGzTl2l6e4jCYd_Rr5aoQ8,1330
 truefoundry/deploy/lib/win32.py,sha256=1RcvPTdlOAJ48rt8rCbE2Ufha2ztRqBAE9dueNXArrY,5009
-truefoundry/deploy/python_deploy_codegen.py,sha256=Ok7ufDY2x3aMJv9KpaRqxiS-ZI-kxBWauIUHst-ug7E,4020
+truefoundry/deploy/python_deploy_codegen.py,sha256=z9VSETb3Lrqn7sUD75EksbmA1vRiEl0LNnz9PTqF8ZM,6462
 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=HIcY3SzQ5lWl7avuuKi3J0Z-PBES6Sf4hgMK-m6_53U,11990
-truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=hgAhd1EGwFLz319Vs-WNXHDJmbKjdgkGPzDnBD1Up1k,12579
+truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=WhlrBuRf7r83qoQdTZSQzHt635fw9D4_qQIqusFWGag,12372
 truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=MROgMxhn9hDEAKwJSWl3iz12tUVvRKzEtqF2QUT6dAk,3343
 truefoundry/deploy/v2/lib/models.py,sha256=pSolLMTArDuYpeNsmeeS5DWliloN_iCDfZSpRllMHUg,1120
 truefoundry/deploy/v2/lib/patched_models.py,sha256=sokVDUdnhe3qx6dXlHM0shbf6HvSlF72-mvi8Lzt_Y8,13968
@@ -125,16 +127,16 @@ truefoundry/langchain/utils.py,sha256=PGLDe9chZ3BuUjakexOGpIqZRFoHEgu-zJ9yKdpLLm
 truefoundry/logger.py,sha256=7dLqW_Q2rEgo-_z1WZnQbGHaoy1L1MP3NqGgssaSS6o,685
 truefoundry/ml/__init__.py,sha256=2A1l7pgqbVRt3cRW_0Lxg92hyJEkMxkCUh1EFprrmc0,942
 truefoundry/ml/artifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=dA_QYAyNCcCuOw5eVxPbU5OeMK8YBNh42vBnGf6I0Sw,44911
+truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=VU8i3jnY62MLfzA3rxXuUjdqLz8Yaw4zqqPWSsf0mBg,45850
 truefoundry/ml/autogen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/autogen/client/__init__.py,sha256=M68r1z8Mi6dEFFKDsWPJ9NGdNOYdm6pcdH7_srCwkCI,16651
+truefoundry/ml/autogen/client/__init__.py,sha256=B6Z0oUmE-S9YqUCDRpKDuTyNWX2Fmx4es7nWx7Bhy10,17600
 truefoundry/ml/autogen/client/api/__init__.py,sha256=3sMSljMIS3UHYeF0BcNvrPPx6VbBSRt_1IfDn-13Kyc,752
 truefoundry/ml/autogen/client/api/auth_api.py,sha256=zpWzJhUmW6HHMY_atlUf0B25k77E1kue2hmix5I5Ih0,7017
 truefoundry/ml/autogen/client/api/deprecated_api.py,sha256=JCQ39Y3VHdgJ1zM4XVabdTl6QpOHtQFjh04XUb6mN9A,24218
-truefoundry/ml/autogen/client/api/experiments_api.py,sha256=vzlPMzclwk7oD2ekUBInIAHLNO3qCAPF38AAvxfDBSY,80682
+truefoundry/ml/autogen/client/api/experiments_api.py,sha256=mRKS8qGzcFJUpTWjfQoFYm2KI8jTHxBzA0l_foSCLM0,74880
 truefoundry/ml/autogen/client/api/health_api.py,sha256=IAPhRAo9CLUT5ipVR1fCf-qDx57UR0wg5ekhtUl8lug,11554
 truefoundry/ml/autogen/client/api/metrics_api.py,sha256=q3L38eD-2hu4_9YvcSdnWDYXD2V8il-X9reinOAABek,14908
-truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py,sha256=Le1fSNI8NwsYKhs-LTpH1fVuTP5YUp99D3WbZkQeo_8,313447
+truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py,sha256=smLST3wBZnMw0VrTDR5y1IuJvzV6n89nLWqUMB5mavY,313547
 truefoundry/ml/autogen/client/api/python_deployment_config_api.py,sha256=8P53aegaaZp1LEV9_GDNSCofgxRXP3Atyprc2bUvvpI,8348
 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
@@ -142,7 +144,7 @@ truefoundry/ml/autogen/client/api_client.py,sha256=8qg-WpadDuKgbRt5yABJ4wVS4IRxd
 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=9zOPn2LD-j3EbZmtt8ShBU8uGMRVpTW2-kvooBG9feE,15267
+truefoundry/ml/autogen/client/models/__init__.py,sha256=PNukskNa-cv5uP25VmJXrZAZygCoyPis4LkMVLRA7M4,16216
 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=rU0h96pEE8K1Ukw2pzDSjq0e6BgtDEuOctI-aZMrpUY,2653
 truefoundry/ml/autogen/client/models/agent.py,sha256=fnMWdEPe5Iw50WKydtu7QAxU419j3ju1IukpChUnGqY,3871
@@ -150,11 +152,12 @@ truefoundry/ml/autogen/client/models/agent_app.py,sha256=h-0xdTMDE6FPzvksnbhVSXO
 truefoundry/ml/autogen/client/models/agent_open_api_tool.py,sha256=QotUMmGqAMpNsRFwVoNJHg9OeobimS-Wc9Eb85VRD24,5023
 truefoundry/ml/autogen/client/models/agent_open_api_tool_with_fqn.py,sha256=8iHIEpGtQJO8QZS66T0FysqKg1VaxJT44hPu_a6zuwE,5121
 truefoundry/ml/autogen/client/models/agent_with_fqn.py,sha256=_LFRmdT0KNe1CRNBfEt9gFbBvGMiwPInbesHSKrLH-g,3976
-truefoundry/ml/autogen/client/models/artifact_dto.py,sha256=I-5m316b3wiSxGrgx7dEymTr--Q8u0jIkl53EbeO4ms,3704
+truefoundry/ml/autogen/client/models/artifact_dto.py,sha256=JWL8_sITJJXbxywcfRj4xqt8jR9pyBmA4TfwNTdX6CE,4243
 truefoundry/ml/autogen/client/models/artifact_response_dto.py,sha256=T_HLjkHnqiO1bnSXiCUOlwDEHGz89mlUNhBq0SJTapo,2226
 truefoundry/ml/autogen/client/models/artifact_type.py,sha256=LBXGYUoesmL5J-0xzCR8jBST3bjb6HY86RmqSGvytRs,828
-truefoundry/ml/autogen/client/models/artifact_version_dto.py,sha256=MdpHGU69R7_ej6_2ppp1a1WWkYcPCdXRyovpQN-WsDs,4868
+truefoundry/ml/autogen/client/models/artifact_version_dto.py,sha256=oGWsRSxLDy8aGMI9dfaaNZH9JpPnqLevWSRmMCsHo44,6233
 truefoundry/ml/autogen/client/models/artifact_version_response_dto.py,sha256=JmiSYdPYG-ki6LCBeF7NtihdQ2LPiHld8cSPivY2GmQ,2421
+truefoundry/ml/autogen/client/models/artifact_version_serialization_format.py,sha256=2Qfgq1nykn0_mKNEJzOxhQwHm_95WIr9wJcHLeFWXQM,774
 truefoundry/ml/autogen/client/models/artifact_version_status.py,sha256=iIcomqhuaJkZH2rT3e6IjQutM79t9hEshW-RCIMKHEY,780
 truefoundry/ml/autogen/client/models/assistant_message.py,sha256=rmCQ4OAbVpjbX3rXiNtv2ePxV-7kq6Eg1Bs_tm_rmNE,2845
 truefoundry/ml/autogen/client/models/authorize_user_for_model_request_dto.py,sha256=G00T6eroxZrDw782-UBCQxlkwkxyAPfmvd9QNcMiZDI,2017
@@ -171,7 +174,7 @@ truefoundry/ml/autogen/client/models/content2_any_of_inner.py,sha256=yeGDgvaMMny
 truefoundry/ml/autogen/client/models/create_artifact_request_dto.py,sha256=U9oPwVw_G_3IIQzV_OqjuL_6ytvmfUwOMAyrYi_UhRo,2217
 truefoundry/ml/autogen/client/models/create_artifact_response_dto.py,sha256=mRRGlXQXW2d3oWv9YOqKG55GHuIpWslSzosUxaNXYXo,1849
 truefoundry/ml/autogen/client/models/create_artifact_version_request_dto.py,sha256=DTmslB75W6Cr_ec1ddEt40VaKowaZVv2ypf7sRH3elI,2284
-truefoundry/ml/autogen/client/models/create_artifact_version_response_dto.py,sha256=iLPOAM7bw-t6hlKL4_mtw8m0QfWj4rRT8vxLdSHE9-8,1905
+truefoundry/ml/autogen/client/models/create_artifact_version_response_dto.py,sha256=rInMtdhkkWeP-3AM5tnyvTkIUIR8fCEnaexQoB4fRmM,2108
 truefoundry/ml/autogen/client/models/create_dataset_request_dto.py,sha256=6Ke3swxo9DG-mbOZOTXosepZ6toHPmYc4GOCtahUs2A,2308
 truefoundry/ml/autogen/client/models/create_experiment_request_dto.py,sha256=yD3u5li_ShKbX1RGomEOkSEXQPZKxqhJGYXAMZTKXDU,3013
 truefoundry/ml/autogen/client/models/create_experiment_response_dto.py,sha256=GmN5gDzieW8xT2RXOhf0QJ5_AL7JzCSwFYJMnys1pgA,1931
@@ -181,9 +184,9 @@ truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_respon
 truefoundry/ml/autogen/client/models/create_multi_part_upload_request_dto.py,sha256=nGrPRw5R44APVGeyegxFLxfi8KAHX-NDORtS1iBw5Xc,2213
 truefoundry/ml/autogen/client/models/create_python_deployment_config_request_dto.py,sha256=1xmhL67Pz0Ckgfky3SIwtD0ayPJA-Vvv51Qec0LVDAA,2232
 truefoundry/ml/autogen/client/models/create_python_deployment_config_response_dto.py,sha256=5fnCJ7ybEcRA03s8ZLGdt1w8_Xz5nt6-PPmvA6Yxt60,2051
-truefoundry/ml/autogen/client/models/create_run_request_dto.py,sha256=6lwK8CLGgAnNtvBn3_0jcKicsWI3DMUfxj1OUctzDwA,2994
+truefoundry/ml/autogen/client/models/create_run_request_dto.py,sha256=OqhHezsPGygl5m3BlEHmKGlCNf2t5FUFQJLwdmHfggg,2841
 truefoundry/ml/autogen/client/models/create_run_response_dto.py,sha256=cFzIkhbiB_IEmQSx1asNAzPMN-MzCBtqWj2PJ3oTLTM,2169
-truefoundry/ml/autogen/client/models/dataset_dto.py,sha256=2GqBLyIFXjNJ8stQRQgB1GoC9eqqArKlBafIGtDOie0,3561
+truefoundry/ml/autogen/client/models/dataset_dto.py,sha256=V3dIdK93YQ8mtolY8Onz2N17HQYm3bLUMn98dVg7tGo,4100
 truefoundry/ml/autogen/client/models/dataset_response_dto.py,sha256=5h3eHpiG4Zzqn5Hm0WpZgyXGjxujKem42_8YnGq5KLo,2205
 truefoundry/ml/autogen/client/models/delete_artifact_versions_request_dto.py,sha256=JF2dQuEe_W1c2ox9zZDER_pZSTbKwSnns8hIfG5o7oA,1905
 truefoundry/ml/autogen/client/models/delete_dataset_request_dto.py,sha256=bfzzgBoekOBcGg2Uzzel80rf2Z5fDgggShkYd0ZlbSM,2143
@@ -191,14 +194,16 @@ truefoundry/ml/autogen/client/models/delete_files_for_dataset_request_dto.py,sha
 truefoundry/ml/autogen/client/models/delete_model_version_request_dto.py,sha256=6abYuueylSgv7TLkIHq8X0NWeFr-J5k72caAyOnn6fw,1873
 truefoundry/ml/autogen/client/models/delete_run_request.py,sha256=jXQmROKfkPA8idoxkUXT-2O6QJ4OgxgFX-CEmjryIpY,1793
 truefoundry/ml/autogen/client/models/delete_tag_request_dto.py,sha256=eQSPMamCJbsehcwWoMxQxlzZKQfX5GEwm3KHGF2Ajts,1901
-truefoundry/ml/autogen/client/models/experiment_dto.py,sha256=LBR3FkOw5j_2w1vrY3PYdqFWJwVXAE-Len8L8ua0XzI,4409
+truefoundry/ml/autogen/client/models/experiment_dto.py,sha256=bl4-Hp1EHl8CEKgWq6sjJEGsTYG3kWQaO-AlskslXKQ,5035
 truefoundry/ml/autogen/client/models/experiment_id_request_dto.py,sha256=ZMH827n_UTpDI30UnkuOam-4ANBKCDgocIzI8StxFR8,1891
 truefoundry/ml/autogen/client/models/experiment_response_dto.py,sha256=wuflV6_f8PQq061-wU2GzNY4BZi8SG8ARCIbSQN1oT4,2268
 truefoundry/ml/autogen/client/models/experiment_tag_dto.py,sha256=nEpCkeZ9ficIDkjmmLfkJeNNokd-Rhgr-cepPWG6L3M,1902
+truefoundry/ml/autogen/client/models/external_model_source.py,sha256=Vzn1sbjMtQYuMEVjd9hxH0TupGG7vwzqadwE4l9vzs8,2286
 truefoundry/ml/autogen/client/models/feature_dto.py,sha256=XM7fF71P_bYP3UlO1ILjnD3DJNpf2tGwAWbrkQyvyA4,1924
 truefoundry/ml/autogen/client/models/feature_value_type.py,sha256=3ZKxxEfrrWk8ePlV7n11wa-FdDH0SRLyHRqSvIU7F-M,732
 truefoundry/ml/autogen/client/models/file_info_dto.py,sha256=7oc8venicsFVk8zT9wHNhHnZGtFkFlqimFnS7ozGL9k,2156
-truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py,sha256=P20UWglW6pzhNJVchC37ypZSHUxzzxf9tfKJkLgxoaY,3361
+truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py,sha256=tArvzRPp80PpfJ4CoMwwaKqPAfzonq-8C-yKBEjs9co,3877
+truefoundry/ml/autogen/client/models/framework.py,sha256=Z_91ZrPWB22mHnKWBgbXHhY5NzHdmfTATBd2iQPNIaU,5131
 truefoundry/ml/autogen/client/models/get_experiment_response_dto.py,sha256=FhJpoUeRbZV1s1KR3Y07Kv6U4enLbbgdpGbguT9us0A,2923
 truefoundry/ml/autogen/client/models/get_latest_run_log_response_dto.py,sha256=g5NG252VzjGHiA-w5nhxmZxGrowJO9XCHDQTjf8y2KE,2266
 truefoundry/ml/autogen/client/models/get_metric_history_response.py,sha256=rTchO8o4sjGsNaZj8TW-VGtGGxsOBjUGzgaAWNtBZ-k,2428
@@ -216,6 +221,7 @@ truefoundry/ml/autogen/client/models/image_content_part.py,sha256=_DXwAxeuagwRWM
 truefoundry/ml/autogen/client/models/image_url.py,sha256=TN8iEdBUmtKdTFlApUodaC4o6X_Ggd3OSOKVPv7laJE,2170
 truefoundry/ml/autogen/client/models/internal_metadata.py,sha256=6OizrIzA3QxlEnwNFXCZGOMJuUDSYHctl_30q27uBDI,6232
 truefoundry/ml/autogen/client/models/latest_run_log_dto.py,sha256=wE8T8bANTb9u14Jv7DNyKWekZiUAvrzvTcE_H1cRhn4,2326
+truefoundry/ml/autogen/client/models/library_name.py,sha256=JKmQMgEj6RcmSv68THAcs1gmh3P3CNrgJ_Rh5VWu9Fs,823
 truefoundry/ml/autogen/client/models/list_artifact_versions_request_dto.py,sha256=nODOEA-tT8XEJKgUXXjvlzT5ecEwz1b9zu-l3RgQl1o,3622
 truefoundry/ml/autogen/client/models/list_artifact_versions_response_dto.py,sha256=G560h_rGn6-X-cxpuddGU8Fp2eBKXvC3RWcJZdZQJBE,2940
 truefoundry/ml/autogen/client/models/list_artifacts_request_dto.py,sha256=o3Jz3VsAa3aJB3YZ6NI6t8TLLoMGeg64jYYZibeo3Oc,2982
@@ -237,7 +243,6 @@ truefoundry/ml/autogen/client/models/list_models_request_dto.py,sha256=YcHYIbHPa
 truefoundry/ml/autogen/client/models/list_models_response_dto.py,sha256=O2QkHGlLPtuGs7ggFuPEf5_hukS1B_Q5fMNPlPClrr4,2662
 truefoundry/ml/autogen/client/models/list_run_artifacts_response_dto.py,sha256=gZrv8iVQZcBcS5PXhABJblvm--E8BY3chJVuCU6GNvI,2715
 truefoundry/ml/autogen/client/models/list_run_logs_response_dto.py,sha256=ckQBBYHVyfSyFnYiczvVXTsnahKSur-2U6wsb-owpEg,2485
-truefoundry/ml/autogen/client/models/list_seed_experiments_response_dto.py,sha256=rsbPpoAQOIOK-jFsfH2Tn30l-tGRLvx_kylN9FSI7n4,2566
 truefoundry/ml/autogen/client/models/log_batch_request_dto.py,sha256=xDas-pmJP5T-8zKd17BbSnglZuX15ei3n9blBZUxa-Y,3702
 truefoundry/ml/autogen/client/models/log_metric_request_dto.py,sha256=QED1aIeaZJG-8iKFxT3NX9ozlUz_wom1xDsyZgL-a-4,2442
 truefoundry/ml/autogen/client/models/log_param_request_dto.py,sha256=Hd7AqrVnSkwfa9CcDbATERaim5Hf8rM59WZQNxSlJzk,2160
@@ -246,10 +251,11 @@ truefoundry/ml/autogen/client/models/metric_collection_dto.py,sha256=02ZWKm2_8Dj
 truefoundry/ml/autogen/client/models/metric_dto.py,sha256=Xrcc8XRY0sZeg_cnb6gyf-AUUIS0FPxSGTCbDzcBzXQ,2174
 truefoundry/ml/autogen/client/models/mime_type.py,sha256=A-N5-Mof_IyHWguXmG67k9wuaS5xwLPcj1J11jrEGyc,897
 truefoundry/ml/autogen/client/models/model_configuration.py,sha256=XKhwqO8caQ8w9XRCdb_pAl-84cd_PWkEutjLNSby6hQ,3861
-truefoundry/ml/autogen/client/models/model_dto.py,sha256=iz9RAxADkeWwe0RmsxiLcYHZ5rZhkRF0PRcy004RT_I,4146
+truefoundry/ml/autogen/client/models/model_dto.py,sha256=TUO74MDqe8XCVJBB2O7TiqiRON8qbE7gi8LNA3uWjYA,4685
 truefoundry/ml/autogen/client/models/model_response_dto.py,sha256=osrTxfygkuhxWj6SkRBALrSnFVPH4LSK6qTufTeZuJg,2163
 truefoundry/ml/autogen/client/models/model_schema_dto.py,sha256=ElEPK7fwuf3eiohNBBHJyl-nhNqzGnDLzZVzuQl-xC0,2620
-truefoundry/ml/autogen/client/models/model_version_dto.py,sha256=1NqfoCKN-MWeWi8_ih4EHfknJnOyUuBbhCu-Kg5WVDE,5917
+truefoundry/ml/autogen/client/models/model_version_dto.py,sha256=6kI1Z2czYAieJZBaVvcV8Y2eeNEUDyjPuTLHTdyWnOE,7282
+truefoundry/ml/autogen/client/models/model_version_manifest.py,sha256=yJN2C_ETddLAeSR4wDuzV3--F2MyiN-XzIJlNayG6Ok,3945
 truefoundry/ml/autogen/client/models/model_version_response_dto.py,sha256=D6XOCyggxqTkbePuypqYSHYh1PYeDv7R_J32q61sDM0,2320
 truefoundry/ml/autogen/client/models/multi_part_upload_dto.py,sha256=Ckq405vud8RQmMyKCJQJBlW5iXO7Y2mlgo8eVkiMvxg,3738
 truefoundry/ml/autogen/client/models/multi_part_upload_response_dto.py,sha256=VjH0kvl7rMjgDHjYGHnsh7KsZ5-qn-k3ksdGLJ49nIM,2431
@@ -263,7 +269,7 @@ truefoundry/ml/autogen/client/models/resolve_agent_app_response_dto.py,sha256=sg
 truefoundry/ml/autogen/client/models/restore_run_request_dto.py,sha256=7iBbil4YfaOHJwP0ifPHp-D1Kt-ZiSRjAaVuU7_idhw,1825
 truefoundry/ml/autogen/client/models/run_data_dto.py,sha256=iPFm5f4x2ymRqea6EDZlcjWYN_-f3t9JZckwskXSTKY,3533
 truefoundry/ml/autogen/client/models/run_dto.py,sha256=inAYRSKnHSkF3cdf2eQYp5wVnOgB47ZPdavxCHXh9zY,2535
-truefoundry/ml/autogen/client/models/run_info_dto.py,sha256=HJqhChLit5ub3LDQarqDQxgd9z2YCZyO06tIt-s-RQ8,3082
+truefoundry/ml/autogen/client/models/run_info_dto.py,sha256=RCGhFUG5kSIcU62LrqXMqh4ediFyeBziV-Z2532Wcmo,3617
 truefoundry/ml/autogen/client/models/run_log_dto.py,sha256=VEIXYILD4AW0QAmokyzS9mo4JmiAV7NgU5TRgymQMek,2529
 truefoundry/ml/autogen/client/models/run_log_input_dto.py,sha256=xQ854-3vcocDzcNYejYVlo2BmEpDLlHh1MbiP2OHen4,2390
 truefoundry/ml/autogen/client/models/run_response_dto.py,sha256=iayBnhwmByVC3iiw_ZUbB_V6FTadEQu9_U2JOs5kvkw,2121
@@ -273,24 +279,31 @@ truefoundry/ml/autogen/client/models/search_runs_response_dto.py,sha256=u0l11v7p
 truefoundry/ml/autogen/client/models/set_experiment_tag_request_dto.py,sha256=nrmi_NxLD1fI2gwlpdqFSMnBS11gRkjS4_GQFHgBcXs,2118
 truefoundry/ml/autogen/client/models/set_tag_request_dto.py,sha256=IRgAdMcWBxmjNV6nZJej4pcNfLmZwrelEZ3otwt7eeE,2144
 truefoundry/ml/autogen/client/models/signed_url_dto.py,sha256=9oHoXBj07xTdc04rqOqJO3eOjQWXCyWPhfHg-6qX60w,1897
+truefoundry/ml/autogen/client/models/source.py,sha256=OqcgjSiQwji_jpfcL29HKlAebURPChuQPqmQzd7CSPk,5961
 truefoundry/ml/autogen/client/models/stop.py,sha256=UlAjRWI6Gjc92D2UME6qRAIfd6GoDI3VBWYkJqgixmU,4760
 truefoundry/ml/autogen/client/models/store_run_logs_request_dto.py,sha256=Cfm95a9T4fDiARBJ9vd3n9Pv7QI69TSVKWi0AG_cmvs,2591
+truefoundry/ml/autogen/client/models/subject.py,sha256=gG1pRl6lPEI1_td2sb752vJeKMKgLWDaoFwepnS4zIQ,2415
+truefoundry/ml/autogen/client/models/subject_type.py,sha256=TOhgDmWSy67NjBCTSbwtF2-cBBB2-FoLJVFDVpJRrsQ,710
 truefoundry/ml/autogen/client/models/system_message.py,sha256=yxRCJPl36FGv2zrsuHbLown88wmnrIpWlKEVRf9WgFQ,2801
+truefoundry/ml/autogen/client/models/tensorflow_framework.py,sha256=8F0CHJj_J0V57GePBuMjSSLmCRuwIot10ghywlBFIeQ,2251
 truefoundry/ml/autogen/client/models/text.py,sha256=HAIK-w_TxB1kXGvVNK1d2cweoQi-GJ9d-0QSZl21u68,4862
 truefoundry/ml/autogen/client/models/text_content_part.py,sha256=fRQu22tiV3rDKmTgVZSLHX-cMcha3q53W8qWNCk5wmU,2492
+truefoundry/ml/autogen/client/models/transformers_framework.py,sha256=5n4h3IvcCRr4Dca9Sv1EGUp07rEgv3c41U47NmGTsu0,2916
+truefoundry/ml/autogen/client/models/truefoundry_model_source.py,sha256=kKxDCkLKTLho2SkVO0df6DoD8czwKbaacbjNRqBMhgY,2396
 truefoundry/ml/autogen/client/models/update_artifact_version_request_dto.py,sha256=AhXj2afmrKskBi5uMT0-mSywZyz3g17lFE3uMRQ5pEM,2246
 truefoundry/ml/autogen/client/models/update_dataset_request_dto.py,sha256=MF-rjJGUP6sDZVH3xP4Q8AEC9CV2PXEO9cXrR6yK57A,2182
 truefoundry/ml/autogen/client/models/update_experiment_request_dto.py,sha256=QE_kZzTDdiXKtQ2U-zbI_gb5vxS5Yv9YupYvD22RkFs,2198
-truefoundry/ml/autogen/client/models/update_model_version_request_dto.py,sha256=dVP9WW0bQV0Bolj_hVbsQQuUT-8lhJ9ocDWsThfARlI,3073
+truefoundry/ml/autogen/client/models/update_model_version_request_dto.py,sha256=0xWqHHflnpO8FyT79psPiPBIGW6Aiovs-b6H7sjMSDQ,3589
 truefoundry/ml/autogen/client/models/update_run_request_dto.py,sha256=fn4VB4Zlllr6Bt_l3aYkkHxAkU9w7vtcqu4iw8BvQWY,2318
 truefoundry/ml/autogen/client/models/update_run_response_dto.py,sha256=wolIFiihGZCBBkuj7mJEYCmGAOqDLFo_bNLcH4LeXSo,2231
+truefoundry/ml/autogen/client/models/upload_model_source.py,sha256=zn2AigQc8uhJiGXdE57X-K4pLPrIK9TuD8g7zNt09-U,2103
 truefoundry/ml/autogen/client/models/url.py,sha256=zMyOmdVkp1ANnQnc9GrHt42xlVwES7FTjrpOp_OtHqo,4953
 truefoundry/ml/autogen/client/models/user_message.py,sha256=6QOmKT_SDfAVZA8V3Bzofm7_c-Qznn4KsE12NmHG4FM,2783
 truefoundry/ml/autogen/client/models/validation_error.py,sha256=mFjwoc8g2-Usu1HXZhOQKQ4TGvLy4lwCzk8dHrJ69aA,2597
 truefoundry/ml/autogen/client/models/validation_error_loc_inner.py,sha256=nThJ5Gmy8W2Wok-ZOI4sK7uRe1BAkLS0qzq-XZbq8zs,4915
 truefoundry/ml/autogen/client/rest.py,sha256=9goba8qHjQuVx5O_yRaTKu7PvBnb7r7swfy3dwuTEgk,14281
-truefoundry/ml/autogen/client_README.md,sha256=8Pd74fIjHyzidtS6HGpusUgHj__nIdd_W6mBsi7tgRQ,33233
-truefoundry/ml/autogen/entities/artifacts.py,sha256=iIzfMPKLXZpzxu55lG0k3edraaZqXS4DLc6-FMzxk6g,10159
+truefoundry/ml/autogen/client_README.md,sha256=ehPzJCsvWR_ghMgIcP-UXMG4AgqcJI6cbpuz1rcsqME,33871
+truefoundry/ml/autogen/entities/artifacts.py,sha256=yXLrEsX28aCxpgRA8x8YM5M55LLlLiWoJqEWRAZu6_Y,16147
 truefoundry/ml/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/cli/cli.py,sha256=ckBcjUpqfhgrPE1okqT_G2iouOLt-0KjpLhHp2YdVFU,256
 truefoundry/ml/cli/commands/__init__.py,sha256=diDUiRUX4l6TtNLI4iF-ZblczkELM7FRViJ-8gGNJQY,82
@@ -302,35 +315,34 @@ truefoundry/ml/clients/utils.py,sha256=c0LdC8moejs-Zm30hu1sCqifLEmqhdq4SfZ_m0nUI
 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/env_vars.py,sha256=0CqhoV6RgxgPju71AJYvo7crHxVr0vJNzWTiSWKnLcI,212
 truefoundry/ml/exceptions.py,sha256=8aJm2NYtAWWsRLu4MbzaoOqHsQZ6RjOFwBWQWqb6qrc,244
 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/artifacts/artifact.py,sha256=pNgLc6GaQ0UVJ6gaSRzpAW5CmGpJMC4iBgBY5FLlofc,17640
+truefoundry/ml/log_types/artifacts/artifact.py,sha256=n2EwAupOigDcUXQQFIIESRqp7zipLZieec0vLQ2DVRQ,17593
 truefoundry/ml/log_types/artifacts/constants.py,sha256=qKxQ5mMvJE4j83BvGW3qNTKunxCiBg_EEjTdgbgJtyE,1036
-truefoundry/ml/log_types/artifacts/dataset.py,sha256=oXi7itXOe3PrnN4_KY7s1GdhWV1lyvWXRWgY6_vPTAs,12981
-truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=fY74EUeCGdHdlkrp_GMoaZPvl7EQctt8OlRM6yFgLF8,3772
-truefoundry/ml/log_types/artifacts/model.py,sha256=NRIvGX0yCLGnLvZkuCRJGGEuONoUhXZSY2jmSUM3JvI,22162
+truefoundry/ml/log_types/artifacts/dataset.py,sha256=a4dxd2EN8p7Ci-cLGGiDOboN3t0395_XhWE1dmTw1Q4,13112
+truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=_EOlNGMg2PFjlevlD6LEOsaQBBwRWs60PkzJases6tE,3927
+truefoundry/ml/log_types/artifacts/model.py,sha256=cuVvBhCaOy165LxyF3NuY7UH61tgYOqFiVUde4WzMwg,21920
 truefoundry/ml/log_types/artifacts/model_extras.py,sha256=TIE73bLKfwIVzNiVcjmaZ841A70BHBwu4XAM6ZAQRFI,1045
-truefoundry/ml/log_types/artifacts/utils.py,sha256=dwmRBrEmwIktVcA1a3lrsUfmROO7THTLgYwJ4MpSUWU,5885
+truefoundry/ml/log_types/artifacts/utils.py,sha256=TZwa-Nu10lrNfiQZTh2AkudI9zLZnMxIpPUDMZtylsk,6307
 truefoundry/ml/log_types/image/__init__.py,sha256=fcOq8yQnNj1rkLcPeIjLXBpdA1WIeiPsXOlAAvMxx7M,76
 truefoundry/ml/log_types/image/constants.py,sha256=wLtGEOA4T5fZHSlOXPuNDLX3lpbCtwlvGKPFk_1fah0,255
-truefoundry/ml/log_types/image/image.py,sha256=y2FOUVTNGtjzGW5AcaXObIxKQDGr1GoiSYBoRAB7ZaM,12310
+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=oFnXNb2o5fVF0zsnRjvqjSjLaphQWUnQCdw72e2uiMY,7347
+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=I2T8tXeAIWpD8EH05fm80mNyX6cs2S1ORI4qoo0HTpQ,60847
+truefoundry/ml/mlfoundry_api.py,sha256=lVsD7WBqAZMNSEauKeiF1jCg_Fnt6IFEPRr4mwn1IMU,60820
 truefoundry/ml/mlfoundry_run.py,sha256=rNJl130iJkpjW3MNoe5-d_J9VJJQBqWHEJCfYeiZCbE,45123
 truefoundry/ml/run_utils.py,sha256=0W208wSLUrbdfk2pjNcZlkUi9bNxG2JORqoe-5rVqHI,2423
-truefoundry/ml/session.py,sha256=Rvq419hdLNBt0KUzKn7jGvKKg9LI_gafOHcw-lFceAE,5330
+truefoundry/ml/session.py,sha256=F83GTC5WwGBjnJ69Ct8MqMnlutYc56JCc6YhEY1Wl-A,5394
 truefoundry/ml/validation_utils.py,sha256=XBSUd9OoyriWJpT3M5LKz17iWY3yVMr3hM5vdaVjtf0,12082
 truefoundry/pydantic_v1.py,sha256=jSuhGtz0Mbk1qYu8jJ1AcnIDK4oxUsdhALc4spqstmM,345
 truefoundry/version.py,sha256=bqiT4Q-VWrTC6P4qfK43mez-Ppf-smWfrl6DcwV7mrw,137
-truefoundry/workflow/__init__.py,sha256=hkDVrnG95duO-9XEmdjgrsvyhjQUVqyeQ1-dADB1QuE,935
+truefoundry/workflow/__init__.py,sha256=c4daVkQE269b69lZRiRrRn4abAfIhVDN_4Na5MsFfmk,1414
 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
@@ -341,7 +353,7 @@ truefoundry/workflow/map_task.py,sha256=2m3qGXQ90k9LdS45q8dqCCECc3qr8t2m_LMCVd1m
 truefoundry/workflow/python_task.py,sha256=SRXRLC4vdBqGjhkwuaY39LEWN6iPCpJAuW17URRdWTY,1128
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.4.3rc0.dist-info/METADATA,sha256=gtyLuL2SsrUP6uuizszDwgBMW7JTX92jIynL1xNy3oA,3101
-truefoundry-0.4.3rc0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.4.3rc0.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.4.3rc0.dist-info/RECORD,,
+truefoundry-0.4.4.dist-info/METADATA,sha256=6TAi6m39237cMPw6VHhWIod_MYAKTp733cSTpOHIKZs,3098
+truefoundry-0.4.4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+truefoundry-0.4.4.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.4.4.dist-info/RECORD,,

truefoundry/ml/autogen/client/models/list_seed_experiments_response_dto.py

@@ -1,81 +0,0 @@
-# coding: utf-8
-
-"""
-FastAPI
-
-No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
-
-The version of the OpenAPI document: 0.1.0
-Generated by OpenAPI Generator (https://openapi-generator.tech)
-
-Do not edit the class manually.
-"""  # noqa: E501
-
-from __future__ import annotations
-
-import json
-import pprint
-import re  # noqa: F401
-
-from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
-from truefoundry.pydantic_v1 import BaseModel, Field, conlist
-
-
-class ListSeedExperimentsResponseDto(BaseModel):
-    """
-    ListSeedExperimentsResponseDto
-    """
-
-    experiments: conlist(ExperimentDto) = Field(...)
-    __properties = ["experiments"]
-
-    class Config:
-        """Pydantic configuration"""
-
-        allow_population_by_field_name = True
-        validate_assignment = True
-
-    def to_str(self) -> str:
-        """Returns the string representation of the model using alias"""
-        return pprint.pformat(self.dict(by_alias=True))
-
-    def to_json(self) -> str:
-        """Returns the JSON representation of the model using alias"""
-        return json.dumps(self.to_dict())
-
-    @classmethod
-    def from_json(cls, json_str: str) -> ListSeedExperimentsResponseDto:
-        """Create an instance of ListSeedExperimentsResponseDto from a JSON string"""
-        return cls.from_dict(json.loads(json_str))
-
-    def to_dict(self):
-        """Returns the dictionary representation of the model using alias"""
-        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
-        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of each item in experiments (list)
-        _items = []
-        if self.experiments:
-            for _item in self.experiments:
-                if _item:
-                    _items.append(_item.to_dict())
-            _dict["experiments"] = _items
-        return _dict
-
-    @classmethod
-    def from_dict(cls, obj: dict) -> ListSeedExperimentsResponseDto:
-        """Create an instance of ListSeedExperimentsResponseDto from a dict"""
-        if obj is None:
-            return None
-
-        if not isinstance(obj, dict):
-            return ListSeedExperimentsResponseDto.parse_obj(obj)
-
-        _obj = ListSeedExperimentsResponseDto.parse_obj(
-            {
-                "experiments": [
-                    ExperimentDto.from_dict(_item) for _item in obj.get("experiments")
-                ]
-                if obj.get("experiments") is not None
-                else None
-            }
-        )
-        return _obj