truefoundry 0.4.4rc9__py3-none-any.whl → 0.4.4rc10__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

@@ -335,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-0.4.4rc9.dist-info → truefoundry-0.4.4rc10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.4rc9
+Version: 0.4.4rc10
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.4.4rc9.dist-info → truefoundry-0.4.4rc10.dist-info}/RECORD

@@ -127,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=uS4FZq4wUhxB8-BDx--fLYTcsLYL_V9clNtb32G0Wo0,45050
+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
@@ -144,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
@@ -152,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
@@ -173,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
@@ -183,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
@@ -193,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
@@ -218,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
@@ -239,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
@@ -248,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
@@ -265,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
@@ -275,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
@@ -308,23 +319,23 @@ truefoundry/ml/exceptions.py,sha256=8aJm2NYtAWWsRLu4MbzaoOqHsQZ6RjOFwBWQWqb6qrc,
 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=ltHLkJLLeeRO1d6BqoSq3VtizicvyjdG_Hf52wqig7w,17478
+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=ou6hF0RnwQ8clgpWW8SJPOu2YCG-xya51b91pGY-PjY,12823
-truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=fY74EUeCGdHdlkrp_GMoaZPvl7EQctt8OlRM6yFgLF8,3772
-truefoundry/ml/log_types/artifacts/model.py,sha256=Qb0wq2qCbPCk4os74xsJwOXUQc-4sdGsBwKkV9emDsI,21998
+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=UV42BLXf3R9Ql7TBJ4zGUoHq2B4ENEU2jK9aaEniGQc,60923
+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=F83GTC5WwGBjnJ69Ct8MqMnlutYc56JCc6YhEY1Wl-A,5394
@@ -342,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.4rc9.dist-info/METADATA,sha256=yhVxkhDM-AIf-DK8mdAV-Y4PmPgBpIJ3ZVdDmje1VD0,3101
-truefoundry-0.4.4rc9.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.4.4rc9.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.4.4rc9.dist-info/RECORD,,
+truefoundry-0.4.4rc10.dist-info/METADATA,sha256=S9m9mBKH3ow88eMA7hHPOgGYw7AJjfvn4fIZDK-m8cg,3102
+truefoundry-0.4.4rc10.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+truefoundry-0.4.4rc10.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.4.4rc10.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