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

truefoundry/ml/{autogen → _autogen}/client/api/mlfoundry_artifacts_api.py

@@ -15,168 +15,170 @@ import re  # noqa: F401
 import warnings
 from typing import Optional
 
-from truefoundry.ml.autogen.client.api_client import ApiClient
-from truefoundry.ml.autogen.client.api_response import ApiResponse
-from truefoundry.ml.autogen.client.exceptions import (  # noqa: F401
+from truefoundry.ml._autogen.client.api_client import ApiClient
+from truefoundry.ml._autogen.client.api_response import ApiResponse
+from truefoundry.ml._autogen.client.exceptions import (  # noqa: F401
     ApiTypeError,
     ApiValueError,
 )
-from truefoundry.ml.autogen.client.models.apply_request_dto import ApplyRequestDto
-from truefoundry.ml.autogen.client.models.apply_response_dto import ApplyResponseDto
-from truefoundry.ml.autogen.client.models.artifact_response_dto import (
+from truefoundry.ml._autogen.client.models.apply_request_dto import ApplyRequestDto
+from truefoundry.ml._autogen.client.models.apply_response_dto import ApplyResponseDto
+from truefoundry.ml._autogen.client.models.artifact_response_dto import (
     ArtifactResponseDto,
 )
-from truefoundry.ml.autogen.client.models.artifact_type import ArtifactType
-from truefoundry.ml.autogen.client.models.artifact_version_response_dto import (
+from truefoundry.ml._autogen.client.models.artifact_type import ArtifactType
+from truefoundry.ml._autogen.client.models.artifact_version_response_dto import (
     ArtifactVersionResponseDto,
 )
-from truefoundry.ml.autogen.client.models.authorize_user_for_model_request_dto import (
+from truefoundry.ml._autogen.client.models.authorize_user_for_model_request_dto import (
     AuthorizeUserForModelRequestDto,
 )
-from truefoundry.ml.autogen.client.models.authorize_user_for_model_version_request_dto import (
+from truefoundry.ml._autogen.client.models.authorize_user_for_model_version_request_dto import (
     AuthorizeUserForModelVersionRequestDto,
 )
-from truefoundry.ml.autogen.client.models.create_artifact_request_dto import (
+from truefoundry.ml._autogen.client.models.create_artifact_request_dto import (
     CreateArtifactRequestDto,
 )
-from truefoundry.ml.autogen.client.models.create_artifact_response_dto import (
+from truefoundry.ml._autogen.client.models.create_artifact_response_dto import (
     CreateArtifactResponseDto,
 )
-from truefoundry.ml.autogen.client.models.create_artifact_version_request_dto import (
+from truefoundry.ml._autogen.client.models.create_artifact_version_request_dto import (
     CreateArtifactVersionRequestDto,
 )
-from truefoundry.ml.autogen.client.models.create_artifact_version_response_dto import (
+from truefoundry.ml._autogen.client.models.create_artifact_version_response_dto import (
     CreateArtifactVersionResponseDto,
 )
-from truefoundry.ml.autogen.client.models.create_dataset_request_dto import (
+from truefoundry.ml._autogen.client.models.create_dataset_request_dto import (
     CreateDatasetRequestDto,
 )
-from truefoundry.ml.autogen.client.models.create_model_version_request_dto import (
+from truefoundry.ml._autogen.client.models.create_model_version_request_dto import (
     CreateModelVersionRequestDto,
 )
-from truefoundry.ml.autogen.client.models.create_multi_part_upload_for_dataset_request_dto import (
+from truefoundry.ml._autogen.client.models.create_multi_part_upload_for_dataset_request_dto import (
     CreateMultiPartUploadForDatasetRequestDto,
 )
-from truefoundry.ml.autogen.client.models.create_multi_part_upload_for_dataset_response_dto import (
+from truefoundry.ml._autogen.client.models.create_multi_part_upload_for_dataset_response_dto import (
     CreateMultiPartUploadForDatasetResponseDto,
 )
-from truefoundry.ml.autogen.client.models.create_multi_part_upload_request_dto import (
+from truefoundry.ml._autogen.client.models.create_multi_part_upload_request_dto import (
     CreateMultiPartUploadRequestDto,
 )
-from truefoundry.ml.autogen.client.models.dataset_response_dto import DatasetResponseDto
-from truefoundry.ml.autogen.client.models.delete_artifact_versions_request_dto import (
+from truefoundry.ml._autogen.client.models.dataset_response_dto import (
+    DatasetResponseDto,
+)
+from truefoundry.ml._autogen.client.models.delete_artifact_versions_request_dto import (
     DeleteArtifactVersionsRequestDto,
 )
-from truefoundry.ml.autogen.client.models.delete_dataset_request_dto import (
+from truefoundry.ml._autogen.client.models.delete_dataset_request_dto import (
     DeleteDatasetRequestDto,
 )
-from truefoundry.ml.autogen.client.models.delete_files_for_dataset_request_dto import (
+from truefoundry.ml._autogen.client.models.delete_files_for_dataset_request_dto import (
     DeleteFilesForDatasetRequestDto,
 )
-from truefoundry.ml.autogen.client.models.delete_model_version_request_dto import (
+from truefoundry.ml._autogen.client.models.delete_model_version_request_dto import (
     DeleteModelVersionRequestDto,
 )
-from truefoundry.ml.autogen.client.models.export_deployment_files_request_dto import (
+from truefoundry.ml._autogen.client.models.export_deployment_files_request_dto import (
     ExportDeploymentFilesRequestDto,
 )
-from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto import (
+from truefoundry.ml._autogen.client.models.finalize_artifact_version_request_dto import (
     FinalizeArtifactVersionRequestDto,
 )
-from truefoundry.ml.autogen.client.models.get_artifact_version_aliases_response_dto import (
+from truefoundry.ml._autogen.client.models.get_artifact_version_aliases_response_dto import (
     GetArtifactVersionAliasesResponseDto,
 )
-from truefoundry.ml.autogen.client.models.get_signed_url_for_dataset_write_request_dto import (
+from truefoundry.ml._autogen.client.models.get_signed_url_for_dataset_write_request_dto import (
     GetSignedURLForDatasetWriteRequestDto,
 )
-from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_read_request_dto import (
+from truefoundry.ml._autogen.client.models.get_signed_urls_for_artifact_version_read_request_dto import (
     GetSignedURLsForArtifactVersionReadRequestDto,
 )
-from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_read_response_dto import (
+from truefoundry.ml._autogen.client.models.get_signed_urls_for_artifact_version_read_response_dto import (
     GetSignedURLsForArtifactVersionReadResponseDto,
 )
-from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_write_request_dto import (
+from truefoundry.ml._autogen.client.models.get_signed_urls_for_artifact_version_write_request_dto import (
     GetSignedURLsForArtifactVersionWriteRequestDto,
 )
-from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_write_response_dto import (
+from truefoundry.ml._autogen.client.models.get_signed_urls_for_artifact_version_write_response_dto import (
     GetSignedURLsForArtifactVersionWriteResponseDto,
 )
-from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_read_request_dto import (
+from truefoundry.ml._autogen.client.models.get_signed_urls_for_dataset_read_request_dto import (
     GetSignedURLsForDatasetReadRequestDto,
 )
-from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_read_response_dto import (
+from truefoundry.ml._autogen.client.models.get_signed_urls_for_dataset_read_response_dto import (
     GetSignedURLsForDatasetReadResponseDto,
 )
-from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_write_response_dto import (
+from truefoundry.ml._autogen.client.models.get_signed_urls_for_dataset_write_response_dto import (
     GetSignedURLsForDatasetWriteResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_artifact_versions_request_dto import (
+from truefoundry.ml._autogen.client.models.list_artifact_versions_request_dto import (
     ListArtifactVersionsRequestDto,
 )
-from truefoundry.ml.autogen.client.models.list_artifact_versions_response_dto import (
+from truefoundry.ml._autogen.client.models.list_artifact_versions_response_dto import (
     ListArtifactVersionsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_artifacts_request_dto import (
+from truefoundry.ml._autogen.client.models.list_artifacts_request_dto import (
     ListArtifactsRequestDto,
 )
-from truefoundry.ml.autogen.client.models.list_artifacts_response_dto import (
+from truefoundry.ml._autogen.client.models.list_artifacts_response_dto import (
     ListArtifactsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_datasets_request_dto import (
+from truefoundry.ml._autogen.client.models.list_datasets_request_dto import (
     ListDatasetsRequestDto,
 )
-from truefoundry.ml.autogen.client.models.list_datasets_response_dto import (
+from truefoundry.ml._autogen.client.models.list_datasets_response_dto import (
     ListDatasetsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_files_for_artifact_version_request_dto import (
+from truefoundry.ml._autogen.client.models.list_files_for_artifact_version_request_dto import (
     ListFilesForArtifactVersionRequestDto,
 )
-from truefoundry.ml.autogen.client.models.list_files_for_artifact_versions_response_dto import (
+from truefoundry.ml._autogen.client.models.list_files_for_artifact_versions_response_dto import (
     ListFilesForArtifactVersionsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_files_for_dataset_request_dto import (
+from truefoundry.ml._autogen.client.models.list_files_for_dataset_request_dto import (
     ListFilesForDatasetRequestDto,
 )
-from truefoundry.ml.autogen.client.models.list_files_for_dataset_response_dto import (
+from truefoundry.ml._autogen.client.models.list_files_for_dataset_response_dto import (
     ListFilesForDatasetResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_model_version_response_dto import (
+from truefoundry.ml._autogen.client.models.list_model_version_response_dto import (
     ListModelVersionResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_model_versions_request_dto import (
+from truefoundry.ml._autogen.client.models.list_model_versions_request_dto import (
     ListModelVersionsRequestDto,
 )
-from truefoundry.ml.autogen.client.models.list_models_request_dto import (
+from truefoundry.ml._autogen.client.models.list_models_request_dto import (
     ListModelsRequestDto,
 )
-from truefoundry.ml.autogen.client.models.list_models_response_dto import (
+from truefoundry.ml._autogen.client.models.list_models_response_dto import (
     ListModelsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.model_response_dto import ModelResponseDto
-from truefoundry.ml.autogen.client.models.model_version_response_dto import (
+from truefoundry.ml._autogen.client.models.model_response_dto import ModelResponseDto
+from truefoundry.ml._autogen.client.models.model_version_response_dto import (
     ModelVersionResponseDto,
 )
-from truefoundry.ml.autogen.client.models.multi_part_upload_response_dto import (
+from truefoundry.ml._autogen.client.models.multi_part_upload_response_dto import (
     MultiPartUploadResponseDto,
 )
-from truefoundry.ml.autogen.client.models.notify_artifact_version_failure_dto import (
+from truefoundry.ml._autogen.client.models.notify_artifact_version_failure_dto import (
     NotifyArtifactVersionFailureDto,
 )
-from truefoundry.ml.autogen.client.models.resolve_agent_app_response_dto import (
+from truefoundry.ml._autogen.client.models.resolve_agent_app_response_dto import (
     ResolveAgentAppResponseDto,
 )
-from truefoundry.ml.autogen.client.models.update_artifact_version_request_dto import (
+from truefoundry.ml._autogen.client.models.update_artifact_version_request_dto import (
     UpdateArtifactVersionRequestDto,
 )
-from truefoundry.ml.autogen.client.models.update_dataset_request_dto import (
+from truefoundry.ml._autogen.client.models.update_dataset_request_dto import (
     UpdateDatasetRequestDto,
 )
-from truefoundry.ml.autogen.client.models.update_model_version_request_dto import (
+from truefoundry.ml._autogen.client.models.update_model_version_request_dto import (
     UpdateModelVersionRequestDto,
 )
-from truefoundry.ml.autogen.client.models.validate_external_storage_root_request_dto import (
+from truefoundry.ml._autogen.client.models.validate_external_storage_root_request_dto import (
     ValidateExternalStorageRootRequestDto,
 )
-from truefoundry.ml.autogen.client.models.validate_external_storage_root_response_dto import (
+from truefoundry.ml._autogen.client.models.validate_external_storage_root_response_dto import (
     ValidateExternalStorageRootResponseDto,
 )
 from truefoundry.pydantic_v1 import (

truefoundry/ml/{autogen → _autogen}/client/api/run_artifacts_api.py

@@ -15,13 +15,13 @@ import re  # noqa: F401
 import warnings
 from typing import Optional
 
-from truefoundry.ml.autogen.client.api_client import ApiClient
-from truefoundry.ml.autogen.client.api_response import ApiResponse
-from truefoundry.ml.autogen.client.exceptions import (  # noqa: F401
+from truefoundry.ml._autogen.client.api_client import ApiClient
+from truefoundry.ml._autogen.client.api_response import ApiResponse
+from truefoundry.ml._autogen.client.exceptions import (  # noqa: F401
     ApiTypeError,
     ApiValueError,
 )
-from truefoundry.ml.autogen.client.models.list_run_artifacts_response_dto import (
+from truefoundry.ml._autogen.client.models.list_run_artifacts_response_dto import (
     ListRunArtifactsResponseDto,
 )
 from truefoundry.pydantic_v1 import (

truefoundry/ml/{autogen → _autogen}/client/api/runs_api.py

@@ -14,58 +14,58 @@ Do not edit the class manually.
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.api_client import ApiClient
-from truefoundry.ml.autogen.client.api_response import ApiResponse
-from truefoundry.ml.autogen.client.exceptions import (  # noqa: F401
+from truefoundry.ml._autogen.client.api_client import ApiClient
+from truefoundry.ml._autogen.client.api_response import ApiResponse
+from truefoundry.ml._autogen.client.exceptions import (  # noqa: F401
     ApiTypeError,
     ApiValueError,
 )
-from truefoundry.ml.autogen.client.models.create_run_request_dto import (
+from truefoundry.ml._autogen.client.models.create_run_request_dto import (
     CreateRunRequestDto,
 )
-from truefoundry.ml.autogen.client.models.create_run_response_dto import (
+from truefoundry.ml._autogen.client.models.create_run_response_dto import (
     CreateRunResponseDto,
 )
-from truefoundry.ml.autogen.client.models.delete_run_request import DeleteRunRequest
-from truefoundry.ml.autogen.client.models.delete_tag_request_dto import (
+from truefoundry.ml._autogen.client.models.delete_run_request import DeleteRunRequest
+from truefoundry.ml._autogen.client.models.delete_tag_request_dto import (
     DeleteTagRequestDto,
 )
-from truefoundry.ml.autogen.client.models.get_latest_run_log_response_dto import (
+from truefoundry.ml._autogen.client.models.get_latest_run_log_response_dto import (
     GetLatestRunLogResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_latest_run_logs_response_dto import (
+from truefoundry.ml._autogen.client.models.list_latest_run_logs_response_dto import (
     ListLatestRunLogsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_run_logs_response_dto import (
+from truefoundry.ml._autogen.client.models.list_run_logs_response_dto import (
     ListRunLogsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.log_batch_request_dto import (
+from truefoundry.ml._autogen.client.models.log_batch_request_dto import (
     LogBatchRequestDto,
 )
-from truefoundry.ml.autogen.client.models.log_metric_request_dto import (
+from truefoundry.ml._autogen.client.models.log_metric_request_dto import (
     LogMetricRequestDto,
 )
-from truefoundry.ml.autogen.client.models.log_param_request_dto import (
+from truefoundry.ml._autogen.client.models.log_param_request_dto import (
     LogParamRequestDto,
 )
-from truefoundry.ml.autogen.client.models.restore_run_request_dto import (
+from truefoundry.ml._autogen.client.models.restore_run_request_dto import (
     RestoreRunRequestDto,
 )
-from truefoundry.ml.autogen.client.models.run_response_dto import RunResponseDto
-from truefoundry.ml.autogen.client.models.search_runs_request_dto import (
+from truefoundry.ml._autogen.client.models.run_response_dto import RunResponseDto
+from truefoundry.ml._autogen.client.models.search_runs_request_dto import (
     SearchRunsRequestDto,
 )
-from truefoundry.ml.autogen.client.models.search_runs_response_dto import (
+from truefoundry.ml._autogen.client.models.search_runs_response_dto import (
     SearchRunsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.set_tag_request_dto import SetTagRequestDto
-from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
+from truefoundry.ml._autogen.client.models.set_tag_request_dto import SetTagRequestDto
+from truefoundry.ml._autogen.client.models.store_run_logs_request_dto import (
     StoreRunLogsRequestDto,
 )
-from truefoundry.ml.autogen.client.models.update_run_request_dto import (
+from truefoundry.ml._autogen.client.models.update_run_request_dto import (
     UpdateRunRequestDto,
 )
-from truefoundry.ml.autogen.client.models.update_run_response_dto import (
+from truefoundry.ml._autogen.client.models.update_run_response_dto import (
     UpdateRunResponseDto,
 )
 from truefoundry.pydantic_v1 import (

truefoundry/ml/{autogen → _autogen}/client/api_client.py

@@ -23,11 +23,11 @@ from urllib.parse import quote
 
 from dateutil.parser import parse
 
-import truefoundry.ml.autogen.client.models
-from truefoundry.ml.autogen.client import rest
-from truefoundry.ml.autogen.client.api_response import ApiResponse
-from truefoundry.ml.autogen.client.configuration import Configuration
-from truefoundry.ml.autogen.client.exceptions import ApiException, ApiValueError
+import truefoundry.ml._autogen.client.models
+from truefoundry.ml._autogen.client import rest
+from truefoundry.ml._autogen.client.api_response import ApiResponse
+from truefoundry.ml._autogen.client.configuration import Configuration
+from truefoundry.ml._autogen.client.exceptions import ApiException, ApiValueError
 from truefoundry.pydantic_v1 import SecretStr
 
 
@@ -376,7 +376,7 @@ class ApiClient:
             if klass in self.NATIVE_TYPES_MAPPING:
                 klass = self.NATIVE_TYPES_MAPPING[klass]
             else:
-                klass = getattr(truefoundry.ml.autogen.client.models, klass)
+                klass = getattr(truefoundry.ml._autogen.client.models, klass)
 
         if klass in self.PRIMITIVE_TYPES:
             return self.__deserialize_primitive(data, klass)

truefoundry/ml/{autogen → _autogen}/client/configuration.py

@@ -121,7 +121,7 @@ class Configuration:
         """Logging Settings
         """
         self.logger["package_logger"] = logging.getLogger(
-            "truefoundry.ml.autogen.client"
+            "truefoundry.ml._autogen.client"
         )
         self.logger["urllib3_logger"] = logging.getLogger("urllib3")
         self.logger_format = "%(asctime)s %(levelname)s %(message)s"