truefoundry 0.5.10rc1__py3-none-any.whl → 0.5.11rc1__py3-none-any.whl

truefoundry/ml/{autogen → _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 → _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.
@@ -17,7 +17,7 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.ml.autogen.client.models.multi_part_upload_dto import (
+from truefoundry.ml._autogen.client.models.multi_part_upload_dto import (
     MultiPartUploadDto,
 )
 from truefoundry.pydantic_v1 import BaseModel, Field

truefoundry/ml/{autogen → _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 → _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 → _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 → _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.
@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.run_tag_dto import RunTagDto
+from truefoundry.ml._autogen.client.models.run_tag_dto import RunTagDto
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
 
 

truefoundry/ml/{autogen → _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.
@@ -17,7 +17,7 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.ml.autogen.client.models.run_dto import RunDto
+from truefoundry.ml._autogen.client.models.run_dto import RunDto
 from truefoundry.pydantic_v1 import BaseModel, Field
 
 

truefoundry/ml/{autogen → _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,124 @@
+# 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="Name of the data directory"
+    )
+    ml_repo: constr(strict=True) = Field(
+        default=...,
+        description="Name of the ML Repo to create the data directory under",
+    )
+    description: Optional[constr(strict=True, max_length=512)] = Field(
+        default=None, description="Description for the data directory"
+    )
+    metadata: Dict[str, Any] = Field(
+        default=...,
+        description='Key Value metadata. Should be valid JSON. For e.g. `{"business-unit": "sales", "quality": "good", "rating": 4.5}`',
+    )
+    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 → _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,8 +19,11 @@ import re  # noqa: F401
 from datetime import datetime
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
-from truefoundry.ml.autogen.client.models.subject import Subject
+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 → _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.
@@ -17,7 +17,7 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.ml.autogen.client.models.dataset_dto import DatasetDto
+from truefoundry.ml._autogen.client.models.dataset_dto import DatasetDto
 from truefoundry.pydantic_v1 import BaseModel, Field
 
 

truefoundry/ml/{autogen → _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 → _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 → _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 → _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 → _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 → _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 → _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.
@@ -18,8 +18,8 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTagDto
-from truefoundry.ml.autogen.client.models.subject import Subject
+from truefoundry.ml._autogen.client.models.experiment_tag_dto import ExperimentTagDto
+from truefoundry.ml._autogen.client.models.subject import Subject
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
 
 
@@ -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 → _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 → _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.
@@ -17,7 +17,7 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
+from truefoundry.ml._autogen.client.models.experiment_dto import ExperimentDto
 from truefoundry.pydantic_v1 import BaseModel, Field
 
 

truefoundry/ml/{autogen → _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 → _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.
@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.model_server import ModelServer
+from truefoundry.ml._autogen.client.models.model_server import ModelServer
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 

truefoundry/ml/{autogen → _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.
@@ -17,19 +17,18 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validator
 
 
 class ExternalBlobStorageSource(BaseModel):
     """
-    +label=External Blob Storage  # noqa: E501
+    ExternalBlobStorageSource
     """
 
-    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"
+    type: StrictStr = Field(default=..., description="Type of the source")
+    uri: constr(strict=True) = Field(
+        default=...,
+        description="URI referencing a path in the blob storage bucket linked to the MLRepo",
     )
     __properties = ["type", "uri"]
 
@@ -40,6 +39,15 @@ class ExternalBlobStorageSource(BaseModel):
             raise ValueError("must be one of enum values ('external')")
         return value
 
+    @validator("uri")
+    def uri_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if not re.match(r"^(s3|gs|wasbs):\/\/.*", value):
+            raise ValueError(
+                r"must validate the regular expression /^(s3|gs|wasbs):\/\/.*/"
+            )
+        return value
+
     class Config:
         """Pydantic configuration"""
 

truefoundry/ml/{autogen → _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,13 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class FastAIFramework(BaseModel):
     """
-    +docs=FastAI framework for the model version +label=FastAI  # noqa: E501
+    FastAIFramework
     """
 
-    type: StrictStr = Field(
-        default=...,
-        description="+label=Type +usage=Type of the framework +value=fastai",
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     __properties = ["type"]
 
     @validator("type")