truefoundry 0.5.9__py3-none-any.whl → 0.5.10rc2__py3-none-any.whl

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -18,7 +18,6 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr
 
 
@@ -32,7 +31,7 @@ class CreateModelVersionRequestDto(BaseModel):
     artifact_metadata: Optional[Dict[str, Any]] = None
     data_path: Optional[StrictStr] = None
     step: Optional[StrictInt] = None
-    internal_metadata: Optional[InternalMetadata] = None
+    internal_metadata: Optional[Dict[str, Any]] = None
     __properties = [
         "artifact_version_id",
         "description",
@@ -64,9 +63,6 @@ class CreateModelVersionRequestDto(BaseModel):
     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 internal_metadata
-        if self.internal_metadata:
-            _dict["internal_metadata"] = self.internal_metadata.to_dict()
         return _dict
 
     @classmethod
@@ -85,11 +81,7 @@ class CreateModelVersionRequestDto(BaseModel):
                 "artifact_metadata": obj.get("artifact_metadata"),
                 "data_path": obj.get("data_path"),
                 "step": obj.get("step"),
-                "internal_metadata": InternalMetadata.from_dict(
-                    obj.get("internal_metadata")
-                )
-                if obj.get("internal_metadata") is not None
-                else None,
+                "internal_metadata": obj.get("internal_metadata"),
             }
         )
         return _obj

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -0,0 +1,126 @@
+# coding: utf-8
+
+"""
+TrueFoundry ML API
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 1.0.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 typing import Any, Dict, Optional
+
+from truefoundry.ml.autogen.client.models.source1 import Source1
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validator
+
+
+class DataDirectoryManifest(BaseModel):
+    """
+    Data Directory manifest.  # noqa: E501
+    """
+
+    type: Optional[StrictStr] = "data-dir"
+    name: constr(strict=True, max_length=256) = Field(
+        default=...,
+        description="+sort=1 +label=Name +usage=Name of the data directory +message=The data directory name should start with lowercase alphabets and can contain alphanumeric and can include '-' in between",
+    )
+    ml_repo: constr(strict=True) = Field(
+        default=...,
+        description="+label=MLRepo +usage=Name of the ML Repo to create the data directory under +uiType=Hidden",
+    )
+    description: Optional[constr(strict=True, max_length=512)] = Field(
+        default=None,
+        description="+sort=2 +label=Description +usage=Description for the data directory",
+    )
+    metadata: Dict[str, Any] = Field(
+        default=...,
+        description='+label=Metadata +usage=Key Value metadata. Should be valid JSON. For e.g. `{"business-unit": "sales", "quality": "good", "rating": 4.5}` +uiType=JsonInput',
+    )
+    source: Source1 = Field(...)
+    __properties = ["type", "name", "ml_repo", "description", "metadata", "source"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in ("data-dir",):
+            raise ValueError("must be one of enum values ('data-dir')")
+        return value
+
+    @validator("name")
+    def name_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
+            )
+        return value
+
+    @validator("ml_repo")
+    def ml_repo_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$/"
+            )
+        return value
+
+    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) -> DataDirectoryManifest:
+        """Create an instance of DataDirectoryManifest 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 source
+        if self.source:
+            _dict["source"] = self.source.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> DataDirectoryManifest:
+        """Create an instance of DataDirectoryManifest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return DataDirectoryManifest.parse_obj(obj)
+
+        _obj = DataDirectoryManifest.parse_obj(
+            {
+                "type": obj.get("type") if obj.get("type") is not None else "data-dir",
+                "name": obj.get("name"),
+                "ml_repo": obj.get("ml_repo"),
+                "description": obj.get("description"),
+                "metadata": obj.get("metadata"),
+                "source": Source1.from_dict(obj.get("source"))
+                if obj.get("source") is not None
+                else None,
+            }
+        )
+        return _obj

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -19,6 +19,9 @@ import re  # noqa: F401
 from datetime import datetime
 from typing import Any, Dict, Optional
 
+from truefoundry.ml.autogen.client.models.data_directory_manifest import (
+    DataDirectoryManifest,
+)
 from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
 from truefoundry.ml.autogen.client.models.subject import Subject
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
@@ -40,6 +43,7 @@ class DatasetDto(BaseModel):
     updated_at: Optional[datetime] = None
     storage_root: Optional[StrictStr] = None
     dataset_metadata: Optional[Dict[str, Any]] = None
+    manifest: DataDirectoryManifest = Field(...)
     usage_code_snippet: Optional[StrictStr] = None
     experiment: Optional[ExperimentDto] = None
     __properties = [
@@ -54,6 +58,7 @@ class DatasetDto(BaseModel):
         "updated_at",
         "storage_root",
         "dataset_metadata",
+        "manifest",
         "usage_code_snippet",
         "experiment",
     ]
@@ -83,6 +88,9 @@ class DatasetDto(BaseModel):
         # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of created_by_subject
         if self.created_by_subject:
             _dict["created_by_subject"] = self.created_by_subject.to_dict()
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of manifest
+        if self.manifest:
+            _dict["manifest"] = self.manifest.to_dict()
         # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of experiment
         if self.experiment:
             _dict["experiment"] = self.experiment.to_dict()
@@ -112,6 +120,9 @@ class DatasetDto(BaseModel):
                 "updated_at": obj.get("updated_at"),
                 "storage_root": obj.get("storage_root"),
                 "dataset_metadata": obj.get("dataset_metadata"),
+                "manifest": DataDirectoryManifest.from_dict(obj.get("manifest"))
+                if obj.get("manifest") is not None
+                else None,
                 "usage_code_snippet": obj.get("usage_code_snippet"),
                 "experiment": ExperimentDto.from_dict(obj.get("experiment"))
                 if obj.get("experiment") is not None

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -40,7 +40,7 @@ class ExperimentDto(BaseModel):
     privacy_type: Optional[StrictStr] = "PRIVATE"
     created_at: Optional[StrictInt] = None
     tenant_name: Optional[StrictStr] = None
-    manifest: Optional[Dict[str, Any]] = None
+    manifest: Dict[str, Any] = Field(...)
     num_runs: Optional[StrictInt] = None
     artifact_type_counts: Optional[Dict[str, StrictInt]] = None
     datasets_count: Optional[StrictInt] = None

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -22,14 +22,15 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class ExternalBlobStorageSource(BaseModel):
     """
-    +label=External Blob Storage  # noqa: E501
+    +label=External Blob Storage Source  # noqa: E501
     """
 
     type: StrictStr = Field(
         default=..., description="+label=Type +usage=Type of the source +value=external"
     )
     uri: StrictStr = Field(
-        default=..., description="+label=URI +usage=URI of the Blob Storage source"
+        default=...,
+        description="+label=URI +usage=URI referencing a path in the blob storage bucket linked to the MLRepo",
     )
     __properties = ["type", "uri"]
 

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -22,7 +22,7 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class FastAIFramework(BaseModel):
     """
-    +docs=FastAI framework for the model version +label=FastAI  # noqa: E501
+    +label=FastAI +icon=fastai  # noqa: E501
     """
 
     type: StrictStr = Field(

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.