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

truefoundry/ml/{autogen → _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.
@@ -16,6 +16,7 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
+from datetime import datetime
 from typing import Optional
 
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictBool, StrictInt, StrictStr
@@ -30,7 +31,8 @@ class FileInfoDto(BaseModel):
     is_dir: StrictBool = Field(...)
     file_size: Optional[StrictInt] = None
     signed_url: Optional[StrictStr] = None
-    __properties = ["path", "is_dir", "file_size", "signed_url"]
+    last_modified: Optional[datetime] = None
+    __properties = ["path", "is_dir", "file_size", "signed_url", "last_modified"]
 
     class Config:
         """Pydantic configuration"""
@@ -71,6 +73,7 @@ class FileInfoDto(BaseModel):
                 "is_dir": obj.get("is_dir"),
                 "file_size": obj.get("file_size"),
                 "signed_url": obj.get("signed_url"),
+                "last_modified": obj.get("last_modified"),
             }
         )
         return _obj

truefoundry/ml/{autogen → _autogen}/client/models/finalize_artifact_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,8 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
-from truefoundry.ml.autogen.client.models.manifest import Manifest
+from truefoundry.ml._autogen.client.models.manifest1 import Manifest1
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr
 
 
@@ -32,11 +31,11 @@ class FinalizeArtifactVersionRequestDto(BaseModel):
     run_uuid: Optional[StrictStr] = None
     description: Optional[StrictStr] = None
     artifact_metadata: Optional[Dict[str, Any]] = None
-    internal_metadata: Optional[InternalMetadata] = None
+    internal_metadata: Optional[Dict[str, Any]] = None
     data_path: Optional[StrictStr] = None
     step: Optional[StrictInt] = None
     artifact_size: Optional[StrictInt] = None
-    manifest: Optional[Manifest] = None
+    manifest: Optional[Manifest1] = None
     __properties = [
         "id",
         "run_uuid",
@@ -71,9 +70,6 @@ class FinalizeArtifactVersionRequestDto(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()
         # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of manifest
         if self.manifest:
             _dict["manifest"] = self.manifest.to_dict()
@@ -94,15 +90,11 @@ class FinalizeArtifactVersionRequestDto(BaseModel):
                 "run_uuid": obj.get("run_uuid"),
                 "description": obj.get("description"),
                 "artifact_metadata": obj.get("artifact_metadata"),
-                "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"),
                 "data_path": obj.get("data_path"),
                 "step": obj.get("step"),
                 "artifact_size": obj.get("artifact_size"),
-                "manifest": Manifest.from_dict(obj.get("manifest"))
+                "manifest": Manifest1.from_dict(obj.get("manifest"))
                 if obj.get("manifest") is not None
                 else None,
             }

truefoundry/ml/{autogen → _autogen}/client/models/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.
@@ -18,26 +18,26 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.fast_ai_framework import FastAIFramework
-from truefoundry.ml.autogen.client.models.gluon_framework import GluonFramework
-from truefoundry.ml.autogen.client.models.h2_o_framework import H2OFramework
-from truefoundry.ml.autogen.client.models.keras_framework import KerasFramework
-from truefoundry.ml.autogen.client.models.light_gbm_framework import LightGBMFramework
-from truefoundry.ml.autogen.client.models.onnx_framework import ONNXFramework
-from truefoundry.ml.autogen.client.models.paddle_framework import PaddleFramework
-from truefoundry.ml.autogen.client.models.py_torch_framework import PyTorchFramework
-from truefoundry.ml.autogen.client.models.sklearn_framework import SklearnFramework
-from truefoundry.ml.autogen.client.models.spa_cy_framework import SpaCyFramework
-from truefoundry.ml.autogen.client.models.stats_models_framework import (
+from truefoundry.ml._autogen.client.models.fast_ai_framework import FastAIFramework
+from truefoundry.ml._autogen.client.models.gluon_framework import GluonFramework
+from truefoundry.ml._autogen.client.models.h2_o_framework import H2OFramework
+from truefoundry.ml._autogen.client.models.keras_framework import KerasFramework
+from truefoundry.ml._autogen.client.models.light_gbm_framework import LightGBMFramework
+from truefoundry.ml._autogen.client.models.onnx_framework import ONNXFramework
+from truefoundry.ml._autogen.client.models.paddle_framework import PaddleFramework
+from truefoundry.ml._autogen.client.models.py_torch_framework import PyTorchFramework
+from truefoundry.ml._autogen.client.models.sklearn_framework import SklearnFramework
+from truefoundry.ml._autogen.client.models.spa_cy_framework import SpaCyFramework
+from truefoundry.ml._autogen.client.models.stats_models_framework import (
     StatsModelsFramework,
 )
-from truefoundry.ml.autogen.client.models.tensor_flow_framework import (
+from truefoundry.ml._autogen.client.models.tensor_flow_framework import (
     TensorFlowFramework,
 )
-from truefoundry.ml.autogen.client.models.transformers_framework import (
+from truefoundry.ml._autogen.client.models.transformers_framework import (
     TransformersFramework,
 )
-from truefoundry.ml.autogen.client.models.xg_boost_framework import XGBoostFramework
+from truefoundry.ml._autogen.client.models.xg_boost_framework import XGBoostFramework
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
@@ -65,7 +65,7 @@ FRAMEWORK_ANY_OF_SCHEMAS = [
 
 class Framework(BaseModel):
     """
-    +label=Framework +usage=Framework for the model version
+    Framework for the model version like Transformers, PyTorch, Sklearn, Xgboost etc with framework specific metadata. This will be used to infer model deployment configuration
     """
 
     # data type: TransformersFramework

truefoundry/ml/{autogen → _autogen}/client/models/get_artifact_version_aliases_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/get_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.
@@ -18,8 +18,8 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
-from truefoundry.ml.autogen.client.models.run_info_dto import RunInfoDto
+from truefoundry.ml._autogen.client.models.experiment_dto import ExperimentDto
+from truefoundry.ml._autogen.client.models.run_info_dto import RunInfoDto
 from truefoundry.pydantic_v1 import BaseModel, Field, conlist
 
 

truefoundry/ml/{autogen → _autogen}/client/models/get_latest_run_log_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_log_dto import RunLogDto
+from truefoundry.ml._autogen.client.models.run_log_dto import RunLogDto
 from truefoundry.pydantic_v1 import BaseModel, Field
 
 

truefoundry/ml/{autogen → _autogen}/client/models/get_metric_history_response.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.metric_dto import MetricDto
+from truefoundry.ml._autogen.client.models.metric_dto import MetricDto
 from truefoundry.pydantic_v1 import BaseModel, Field, conlist
 
 

truefoundry/ml/{autogen → _autogen}/client/models/get_signed_url_for_dataset_write_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/get_signed_urls_for_artifact_version_read_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/get_signed_urls_for_artifact_version_read_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.signed_url_dto import SignedURLDto
+from truefoundry.ml._autogen.client.models.signed_url_dto import SignedURLDto
 from truefoundry.pydantic_v1 import BaseModel, Field, conlist
 
 

truefoundry/ml/{autogen → _autogen}/client/models/get_signed_urls_for_artifact_version_write_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/get_signed_urls_for_artifact_version_write_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.signed_url_dto import SignedURLDto
+from truefoundry.ml._autogen.client.models.signed_url_dto import SignedURLDto
 from truefoundry.pydantic_v1 import BaseModel, Field, conlist
 
 

truefoundry/ml/{autogen → _autogen}/client/models/get_signed_urls_for_dataset_read_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/get_signed_urls_for_dataset_read_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.signed_url_dto import SignedURLDto
+from truefoundry.ml._autogen.client.models.signed_url_dto import SignedURLDto
 from truefoundry.pydantic_v1 import BaseModel, Field, conlist
 
 

truefoundry/ml/{autogen → _autogen}/client/models/get_signed_urls_for_dataset_write_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.signed_url_dto import SignedURLDto
+from truefoundry.ml._autogen.client.models.signed_url_dto import SignedURLDto
 from truefoundry.pydantic_v1 import BaseModel, Field, conlist
 
 

truefoundry/ml/{autogen → _autogen}/client/models/get_tenant_id_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/gluon_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,12 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class GluonFramework(BaseModel):
     """
-    +docs=Gluon framework for the model version +label=Gluon  # noqa: E501
+    GluonFramework
     """
 
-    type: StrictStr = Field(
-        default=..., description="+label=Type +usage=Type of the framework +value=gluon"
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     __properties = ["type"]
 
     @validator("type")

truefoundry/ml/{autogen → _autogen}/client/models/h2_o_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,12 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class H2OFramework(BaseModel):
     """
-    +docs=H2O framework for the model version +label=H2O  # noqa: E501
+    H2OFramework
     """
 
-    type: StrictStr = Field(
-        default=..., description="+label=Type +usage=Type of the framework +value=h2o"
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     __properties = ["type"]
 
     @validator("type")

truefoundry/ml/{autogen → _autogen}/client/models/http_validation_error.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.validation_error import ValidationError
+from truefoundry.ml._autogen.client.models.validation_error import ValidationError
 from truefoundry.pydantic_v1 import BaseModel, conlist
 
 

truefoundry/ml/{autogen → _autogen}/client/models/image_content_part.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.ml.autogen.client.models.image_url import ImageUrl
+from truefoundry.ml._autogen.client.models.image_url import ImageUrl
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
 class ImageContentPart(BaseModel):
     """
-    +usage=Image URL +label=Image URL  # noqa: E501
+    Image content for the message  # noqa: E501
     """
 
-    type: StrictStr = Field(...)
+    type: StrictStr = Field(default=..., description="Type of the content part")
     image_url: ImageUrl = Field(
-        default=...,
-        description="+label=URL for the image +usage=The URL for the image, must be a non-empty valid HTTPS URL or a data URL",
+        default=..., description="Image URL linking to the image"
     )
     __properties = ["type", "image_url"]
 

truefoundry/ml/{autogen → _autogen}/client/models/image_url.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,13 +17,13 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.ml.autogen.client.models.url import Url
+from truefoundry.ml._autogen.client.models.url import Url
 from truefoundry.pydantic_v1 import BaseModel, Field
 
 
 class ImageUrl(BaseModel):
     """
-    +label=URL for the image +usage=The URL for the image, must be a non-empty valid HTTPS URL or a data URL  # noqa: E501
+    Image URL linking to the image  # noqa: E501
     """
 
     url: Url = Field(...)

truefoundry/ml/{autogen → _autogen}/client/models/infer_method_name.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,7 +19,7 @@ from aenum import Enum
 
 class InferMethodName(str, Enum):
     """
-    +label=Inference Method Name +usage=Name of the method used for inference
+    Name of the method used for inference
     """
 
     """

truefoundry/ml/{autogen → _autogen}/client/models/keras_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,12 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class KerasFramework(BaseModel):
     """
-    +docs=Keras framework for the model version +label=Keras  # noqa: E501
+    KerasFramework
     """
 
-    type: StrictStr = Field(
-        default=..., description="+label=Type +usage=Type of the framework +value=keras"
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     __properties = ["type"]
 
     @validator("type")

truefoundry/ml/{autogen → _autogen}/client/models/latest_run_log_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_log_dto import RunLogDto
+from truefoundry.ml._autogen.client.models.run_log_dto import RunLogDto
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, conlist