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

truefoundry/ml/_autogen/client/__init__.py

@@ -0,0 +1,457 @@
+# coding: utf-8
+
+# flake8: noqa
+
+"""
+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
+
+__version__ = "0.1.0"
+
+# import apis into sdk package
+from truefoundry.ml._autogen.client.api.auth_api import AuthApi
+from truefoundry.ml._autogen.client.api.deprecated_api import DeprecatedApi
+from truefoundry.ml._autogen.client.api.experiments_api import ExperimentsApi
+from truefoundry.ml._autogen.client.api.generate_code_snippet_api import (
+    GenerateCodeSnippetApi,
+)
+from truefoundry.ml._autogen.client.api.health_api import HealthApi
+from truefoundry.ml._autogen.client.api.metrics_api import MetricsApi
+from truefoundry.ml._autogen.client.api.mlfoundry_artifacts_api import (
+    MlfoundryArtifactsApi,
+)
+from truefoundry.ml._autogen.client.api.run_artifacts_api import RunArtifactsApi
+from truefoundry.ml._autogen.client.api.runs_api import RunsApi
+
+# import ApiClient
+from truefoundry.ml._autogen.client.api_response import ApiResponse
+from truefoundry.ml._autogen.client.api_client import ApiClient
+from truefoundry.ml._autogen.client.configuration import Configuration
+from truefoundry.ml._autogen.client.exceptions import OpenApiException
+from truefoundry.ml._autogen.client.exceptions import ApiTypeError
+from truefoundry.ml._autogen.client.exceptions import ApiValueError
+from truefoundry.ml._autogen.client.exceptions import ApiKeyError
+from truefoundry.ml._autogen.client.exceptions import ApiAttributeError
+from truefoundry.ml._autogen.client.exceptions import ApiException
+
+# import models into sdk package
+from truefoundry.ml._autogen.client.models.agent_app import AgentApp
+from truefoundry.ml._autogen.client.models.agent_manifest import AgentManifest
+from truefoundry.ml._autogen.client.models.agent_open_api_tool_manifest import (
+    AgentOpenAPIToolManifest,
+)
+from truefoundry.ml._autogen.client.models.agent_open_api_tool_with_fqn import (
+    AgentOpenAPIToolWithFQN,
+)
+from truefoundry.ml._autogen.client.models.agent_with_fqn import AgentWithFQN
+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_dto import ArtifactDto
+from truefoundry.ml._autogen.client.models.artifact_manifest import ArtifactManifest
+from truefoundry.ml._autogen.client.models.artifact_path import ArtifactPath
+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_dto import (
+    ArtifactVersionDto,
+)
+from truefoundry.ml._autogen.client.models.artifact_version_response_dto import (
+    ArtifactVersionResponseDto,
+)
+from truefoundry.ml._autogen.client.models.artifact_version_serialization_format import (
+    ArtifactVersionSerializationFormat,
+)
+from truefoundry.ml._autogen.client.models.artifact_version_status import (
+    ArtifactVersionStatus,
+)
+from truefoundry.ml._autogen.client.models.assistant_message import AssistantMessage
+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 (
+    AuthorizeUserForModelVersionRequestDto,
+)
+from truefoundry.ml._autogen.client.models.blob_storage_reference import (
+    BlobStorageReference,
+)
+from truefoundry.ml._autogen.client.models.body_get_search_runs_get import (
+    BodyGetSearchRunsGet,
+)
+from truefoundry.ml._autogen.client.models.chat_prompt_manifest import (
+    ChatPromptManifest,
+)
+from truefoundry.ml._autogen.client.models.chat_prompt_manifest_messages_inner import (
+    ChatPromptManifestMessagesInner,
+)
+from truefoundry.ml._autogen.client.models.columns_dto import ColumnsDto
+from truefoundry.ml._autogen.client.models.command import Command
+from truefoundry.ml._autogen.client.models.content import Content
+from truefoundry.ml._autogen.client.models.content1 import Content1
+from truefoundry.ml._autogen.client.models.content2 import Content2
+from truefoundry.ml._autogen.client.models.content2_any_of_inner import (
+    Content2AnyOfInner,
+)
+from truefoundry.ml._autogen.client.models.create_artifact_request_dto import (
+    CreateArtifactRequestDto,
+)
+from truefoundry.ml._autogen.client.models.create_artifact_response_dto import (
+    CreateArtifactResponseDto,
+)
+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 (
+    CreateArtifactVersionResponseDto,
+)
+from truefoundry.ml._autogen.client.models.create_dataset_request_dto import (
+    CreateDatasetRequestDto,
+)
+from truefoundry.ml._autogen.client.models.create_experiment_request_dto import (
+    CreateExperimentRequestDto,
+)
+from truefoundry.ml._autogen.client.models.create_experiment_response_dto import (
+    CreateExperimentResponseDto,
+)
+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 (
+    CreateMultiPartUploadForDatasetRequestDto,
+)
+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 (
+    CreateMultiPartUploadRequestDto,
+)
+from truefoundry.ml._autogen.client.models.create_python_deployment_config_request_dto import (
+    CreatePythonDeploymentConfigRequestDto,
+)
+from truefoundry.ml._autogen.client.models.create_python_deployment_config_response_dto import (
+    CreatePythonDeploymentConfigResponseDto,
+)
+from truefoundry.ml._autogen.client.models.create_run_request_dto import (
+    CreateRunRequestDto,
+)
+from truefoundry.ml._autogen.client.models.create_run_response_dto import (
+    CreateRunResponseDto,
+)
+from truefoundry.ml._autogen.client.models.create_workflow_task_config_request_dto import (
+    CreateWorkflowTaskConfigRequestDto,
+)
+from truefoundry.ml._autogen.client.models.data_directory_manifest import (
+    DataDirectoryManifest,
+)
+from truefoundry.ml._autogen.client.models.dataset_dto import DatasetDto
+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 (
+    DeleteDatasetRequestDto,
+)
+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 (
+    DeleteModelVersionRequestDto,
+)
+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.experiment_dto import ExperimentDto
+from truefoundry.ml._autogen.client.models.experiment_id_request_dto import (
+    ExperimentIdRequestDto,
+)
+from truefoundry.ml._autogen.client.models.experiment_response_dto import (
+    ExperimentResponseDto,
+)
+from truefoundry.ml._autogen.client.models.experiment_tag_dto import ExperimentTagDto
+from truefoundry.ml._autogen.client.models.export_deployment_files_request_dto import (
+    ExportDeploymentFilesRequestDto,
+)
+from truefoundry.ml._autogen.client.models.external_blob_storage_source import (
+    ExternalBlobStorageSource,
+)
+from truefoundry.ml._autogen.client.models.fast_ai_framework import FastAIFramework
+from truefoundry.ml._autogen.client.models.file_info_dto import FileInfoDto
+from truefoundry.ml._autogen.client.models.finalize_artifact_version_request_dto import (
+    FinalizeArtifactVersionRequestDto,
+)
+from truefoundry.ml._autogen.client.models.framework import Framework
+from truefoundry.ml._autogen.client.models.get_artifact_version_aliases_response_dto import (
+    GetArtifactVersionAliasesResponseDto,
+)
+from truefoundry.ml._autogen.client.models.get_experiment_response_dto import (
+    GetExperimentResponseDto,
+)
+from truefoundry.ml._autogen.client.models.get_latest_run_log_response_dto import (
+    GetLatestRunLogResponseDto,
+)
+from truefoundry.ml._autogen.client.models.get_metric_history_response import (
+    GetMetricHistoryResponse,
+)
+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 (
+    GetSignedURLsForArtifactVersionReadRequestDto,
+)
+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 (
+    GetSignedURLsForArtifactVersionWriteRequestDto,
+)
+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 (
+    GetSignedURLsForDatasetReadRequestDto,
+)
+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 (
+    GetSignedURLsForDatasetWriteResponseDto,
+)
+from truefoundry.ml._autogen.client.models.get_tenant_id_response_dto import (
+    GetTenantIdResponseDto,
+)
+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.http_validation_error import (
+    HTTPValidationError,
+)
+from truefoundry.ml._autogen.client.models.image_content_part import ImageContentPart
+from truefoundry.ml._autogen.client.models.image_url import ImageUrl
+from truefoundry.ml._autogen.client.models.infer_method_name import InferMethodName
+from truefoundry.ml._autogen.client.models.keras_framework import KerasFramework
+from truefoundry.ml._autogen.client.models.latest_run_log_dto import LatestRunLogDto
+from truefoundry.ml._autogen.client.models.library_name import LibraryName
+from truefoundry.ml._autogen.client.models.light_gbm_framework import LightGBMFramework
+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 (
+    ListArtifactVersionsResponseDto,
+)
+from truefoundry.ml._autogen.client.models.list_artifacts_request_dto import (
+    ListArtifactsRequestDto,
+)
+from truefoundry.ml._autogen.client.models.list_artifacts_response_dto import (
+    ListArtifactsResponseDto,
+)
+from truefoundry.ml._autogen.client.models.list_colums_response_dto import (
+    ListColumsResponseDto,
+)
+from truefoundry.ml._autogen.client.models.list_datasets_request_dto import (
+    ListDatasetsRequestDto,
+)
+from truefoundry.ml._autogen.client.models.list_datasets_response_dto import (
+    ListDatasetsResponseDto,
+)
+from truefoundry.ml._autogen.client.models.list_experiments_response_dto import (
+    ListExperimentsResponseDto,
+)
+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 (
+    ListFilesForArtifactVersionsResponseDto,
+)
+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 (
+    ListFilesForDatasetResponseDto,
+)
+from truefoundry.ml._autogen.client.models.list_latest_run_logs_response_dto import (
+    ListLatestRunLogsResponseDto,
+)
+from truefoundry.ml._autogen.client.models.list_metric_history_request_dto import (
+    ListMetricHistoryRequestDto,
+)
+from truefoundry.ml._autogen.client.models.list_metric_history_response_dto import (
+    ListMetricHistoryResponseDto,
+)
+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 (
+    ListModelVersionsRequestDto,
+)
+from truefoundry.ml._autogen.client.models.list_models_request_dto import (
+    ListModelsRequestDto,
+)
+from truefoundry.ml._autogen.client.models.list_models_response_dto import (
+    ListModelsResponseDto,
+)
+from truefoundry.ml._autogen.client.models.list_run_artifacts_response_dto import (
+    ListRunArtifactsResponseDto,
+)
+from truefoundry.ml._autogen.client.models.list_run_logs_response_dto import (
+    ListRunLogsResponseDto,
+)
+from truefoundry.ml._autogen.client.models.local_artifact_source import (
+    LocalArtifactSource,
+)
+from truefoundry.ml._autogen.client.models.local_model_source import LocalModelSource
+from truefoundry.ml._autogen.client.models.log_batch_request_dto import (
+    LogBatchRequestDto,
+)
+from truefoundry.ml._autogen.client.models.log_metric_request_dto import (
+    LogMetricRequestDto,
+)
+from truefoundry.ml._autogen.client.models.log_param_request_dto import (
+    LogParamRequestDto,
+)
+from truefoundry.ml._autogen.client.models.manifest import Manifest
+from truefoundry.ml._autogen.client.models.manifest1 import Manifest1
+from truefoundry.ml._autogen.client.models.manifest2 import Manifest2
+from truefoundry.ml._autogen.client.models.method import Method
+from truefoundry.ml._autogen.client.models.metric_collection_dto import (
+    MetricCollectionDto,
+)
+from truefoundry.ml._autogen.client.models.metric_dto import MetricDto
+from truefoundry.ml._autogen.client.models.mime_type import MimeType
+from truefoundry.ml._autogen.client.models.model_configuration import ModelConfiguration
+from truefoundry.ml._autogen.client.models.model_dto import ModelDto
+from truefoundry.ml._autogen.client.models.model_manifest import ModelManifest
+from truefoundry.ml._autogen.client.models.model_response_dto import ModelResponseDto
+from truefoundry.ml._autogen.client.models.model_server import ModelServer
+from truefoundry.ml._autogen.client.models.model_version_dto import ModelVersionDto
+from truefoundry.ml._autogen.client.models.model_version_environment import (
+    ModelVersionEnvironment,
+)
+from truefoundry.ml._autogen.client.models.model_version_response_dto import (
+    ModelVersionResponseDto,
+)
+from truefoundry.ml._autogen.client.models.multi_part_upload_dto import (
+    MultiPartUploadDto,
+)
+from truefoundry.ml._autogen.client.models.multi_part_upload_response_dto import (
+    MultiPartUploadResponseDto,
+)
+from truefoundry.ml._autogen.client.models.multi_part_upload_storage_provider import (
+    MultiPartUploadStorageProvider,
+)
+from truefoundry.ml._autogen.client.models.notify_artifact_version_failure_dto import (
+    NotifyArtifactVersionFailureDto,
+)
+from truefoundry.ml._autogen.client.models.onnx_framework import ONNXFramework
+from truefoundry.ml._autogen.client.models.openapi_spec import OpenapiSpec
+from truefoundry.ml._autogen.client.models.paddle_framework import PaddleFramework
+from truefoundry.ml._autogen.client.models.param_dto import ParamDto
+from truefoundry.ml._autogen.client.models.parameters import Parameters
+from truefoundry.ml._autogen.client.models.py_torch_framework import PyTorchFramework
+from truefoundry.ml._autogen.client.models.resolve_agent_app_response_dto import (
+    ResolveAgentAppResponseDto,
+)
+from truefoundry.ml._autogen.client.models.restore_run_request_dto import (
+    RestoreRunRequestDto,
+)
+from truefoundry.ml._autogen.client.models.run_data_dto import RunDataDto
+from truefoundry.ml._autogen.client.models.run_dto import RunDto
+from truefoundry.ml._autogen.client.models.run_info_dto import RunInfoDto
+from truefoundry.ml._autogen.client.models.run_log_dto import RunLogDto
+from truefoundry.ml._autogen.client.models.run_log_input_dto import RunLogInputDto
+from truefoundry.ml._autogen.client.models.run_response_dto import RunResponseDto
+from truefoundry.ml._autogen.client.models.run_tag_dto import RunTagDto
+from truefoundry.ml._autogen.client.models.search_runs_request_dto import (
+    SearchRunsRequestDto,
+)
+from truefoundry.ml._autogen.client.models.search_runs_response_dto import (
+    SearchRunsResponseDto,
+)
+from truefoundry.ml._autogen.client.models.set_experiment_tag_request_dto import (
+    SetExperimentTagRequestDto,
+)
+from truefoundry.ml._autogen.client.models.set_tag_request_dto import SetTagRequestDto
+from truefoundry.ml._autogen.client.models.signed_url_dto import SignedURLDto
+from truefoundry.ml._autogen.client.models.sklearn_framework import SklearnFramework
+from truefoundry.ml._autogen.client.models.sklearn_model_schema import (
+    SklearnModelSchema,
+)
+from truefoundry.ml._autogen.client.models.sklearn_serialization_format import (
+    SklearnSerializationFormat,
+)
+from truefoundry.ml._autogen.client.models.source import Source
+from truefoundry.ml._autogen.client.models.source1 import Source1
+from truefoundry.ml._autogen.client.models.source2 import Source2
+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.stop import Stop
+from truefoundry.ml._autogen.client.models.store_run_logs_request_dto import (
+    StoreRunLogsRequestDto,
+)
+from truefoundry.ml._autogen.client.models.subject import Subject
+from truefoundry.ml._autogen.client.models.subject_type import SubjectType
+from truefoundry.ml._autogen.client.models.system_message import SystemMessage
+from truefoundry.ml._autogen.client.models.tensor_flow_framework import (
+    TensorFlowFramework,
+)
+from truefoundry.ml._autogen.client.models.text import Text
+from truefoundry.ml._autogen.client.models.text_content_part import TextContentPart
+from truefoundry.ml._autogen.client.models.transformers_framework import (
+    TransformersFramework,
+)
+from truefoundry.ml._autogen.client.models.trigger_job_run_config_request_dto import (
+    TriggerJobRunConfigRequestDto,
+)
+from truefoundry.ml._autogen.client.models.trigger_job_run_config_response_dto import (
+    TriggerJobRunConfigResponseDto,
+)
+from truefoundry.ml._autogen.client.models.true_foundry_managed_source import (
+    TrueFoundryManagedSource,
+)
+from truefoundry.ml._autogen.client.models.update_artifact_version_request_dto import (
+    UpdateArtifactVersionRequestDto,
+)
+from truefoundry.ml._autogen.client.models.update_dataset_request_dto import (
+    UpdateDatasetRequestDto,
+)
+from truefoundry.ml._autogen.client.models.update_experiment_request_dto import (
+    UpdateExperimentRequestDto,
+)
+from truefoundry.ml._autogen.client.models.update_model_version_request_dto import (
+    UpdateModelVersionRequestDto,
+)
+from truefoundry.ml._autogen.client.models.update_run_request_dto import (
+    UpdateRunRequestDto,
+)
+from truefoundry.ml._autogen.client.models.update_run_response_dto import (
+    UpdateRunResponseDto,
+)
+from truefoundry.ml._autogen.client.models.url import Url
+from truefoundry.ml._autogen.client.models.user_message import UserMessage
+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 (
+    ValidateExternalStorageRootResponseDto,
+)
+from truefoundry.ml._autogen.client.models.validation_error import ValidationError
+from truefoundry.ml._autogen.client.models.validation_error_loc_inner import (
+    ValidationErrorLocInner,
+)
+from truefoundry.ml._autogen.client.models.xg_boost_framework import XGBoostFramework
+from truefoundry.ml._autogen.client.models.xg_boost_model_schema import (
+    XGBoostModelSchema,
+)
+from truefoundry.ml._autogen.client.models.xg_boost_serialization_format import (
+    XGBoostSerializationFormat,
+)

truefoundry/ml/_autogen/client/api/__init__.py

@@ -0,0 +1,16 @@
+# flake8: noqa
+
+# import apis into api package
+from truefoundry.ml._autogen.client.api.auth_api import AuthApi
+from truefoundry.ml._autogen.client.api.deprecated_api import DeprecatedApi
+from truefoundry.ml._autogen.client.api.experiments_api import ExperimentsApi
+from truefoundry.ml._autogen.client.api.generate_code_snippet_api import (
+    GenerateCodeSnippetApi,
+)
+from truefoundry.ml._autogen.client.api.health_api import HealthApi
+from truefoundry.ml._autogen.client.api.metrics_api import MetricsApi
+from truefoundry.ml._autogen.client.api.mlfoundry_artifacts_api import (
+    MlfoundryArtifactsApi,
+)
+from truefoundry.ml._autogen.client.api.run_artifacts_api import RunArtifactsApi
+from truefoundry.ml._autogen.client.api.runs_api import RunsApi

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

@@ -13,13 +13,13 @@ Do not edit the class manually.
 
 import re  # 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
+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.get_tenant_id_response_dto import (
+from truefoundry.ml._autogen.client.models.get_tenant_id_response_dto import (
     GetTenantIdResponseDto,
 )
 from truefoundry.pydantic_v1 import StrictStr, validate_arguments

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

@@ -15,20 +15,20 @@ 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.body_get_search_runs_get import (
+from truefoundry.ml._autogen.client.models.body_get_search_runs_get import (
     BodyGetSearchRunsGet,
 )
-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.run_response_dto import RunResponseDto
-from truefoundry.ml.autogen.client.models.search_runs_response_dto import (
+from truefoundry.ml._autogen.client.models.run_response_dto import RunResponseDto
+from truefoundry.ml._autogen.client.models.search_runs_response_dto import (
     SearchRunsResponseDto,
 )
 from truefoundry.pydantic_v1 import (

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

@@ -15,37 +15,37 @@ 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.create_experiment_request_dto import (
+from truefoundry.ml._autogen.client.models.create_experiment_request_dto import (
     CreateExperimentRequestDto,
 )
-from truefoundry.ml.autogen.client.models.create_experiment_response_dto import (
+from truefoundry.ml._autogen.client.models.create_experiment_response_dto import (
     CreateExperimentResponseDto,
 )
-from truefoundry.ml.autogen.client.models.experiment_id_request_dto import (
+from truefoundry.ml._autogen.client.models.experiment_id_request_dto import (
     ExperimentIdRequestDto,
 )
-from truefoundry.ml.autogen.client.models.experiment_response_dto import (
+from truefoundry.ml._autogen.client.models.experiment_response_dto import (
     ExperimentResponseDto,
 )
-from truefoundry.ml.autogen.client.models.get_experiment_response_dto import (
+from truefoundry.ml._autogen.client.models.get_experiment_response_dto import (
     GetExperimentResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_colums_response_dto import (
+from truefoundry.ml._autogen.client.models.list_colums_response_dto import (
     ListColumsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_experiments_response_dto import (
+from truefoundry.ml._autogen.client.models.list_experiments_response_dto import (
     ListExperimentsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.set_experiment_tag_request_dto import (
+from truefoundry.ml._autogen.client.models.set_experiment_tag_request_dto import (
     SetExperimentTagRequestDto,
 )
-from truefoundry.ml.autogen.client.models.update_experiment_request_dto import (
+from truefoundry.ml._autogen.client.models.update_experiment_request_dto import (
     UpdateExperimentRequestDto,
 )
 from truefoundry.pydantic_v1 import (

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

@@ -13,25 +13,25 @@ Do not edit the class manually.
 
 import re  # 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
+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_python_deployment_config_request_dto import (
+from truefoundry.ml._autogen.client.models.create_python_deployment_config_request_dto import (
     CreatePythonDeploymentConfigRequestDto,
 )
-from truefoundry.ml.autogen.client.models.create_python_deployment_config_response_dto import (
+from truefoundry.ml._autogen.client.models.create_python_deployment_config_response_dto import (
     CreatePythonDeploymentConfigResponseDto,
 )
-from truefoundry.ml.autogen.client.models.create_workflow_task_config_request_dto import (
+from truefoundry.ml._autogen.client.models.create_workflow_task_config_request_dto import (
     CreateWorkflowTaskConfigRequestDto,
 )
-from truefoundry.ml.autogen.client.models.trigger_job_run_config_request_dto import (
+from truefoundry.ml._autogen.client.models.trigger_job_run_config_request_dto import (
     TriggerJobRunConfigRequestDto,
 )
-from truefoundry.ml.autogen.client.models.trigger_job_run_config_response_dto import (
+from truefoundry.ml._autogen.client.models.trigger_job_run_config_response_dto import (
     TriggerJobRunConfigResponseDto,
 )
 from truefoundry.pydantic_v1 import validate_arguments

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

@@ -13,9 +13,9 @@ Do not edit the class manually.
 
 import re  # 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
+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,
 )

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

@@ -14,19 +14,19 @@ 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.get_metric_history_response import (
+from truefoundry.ml._autogen.client.models.get_metric_history_response import (
     GetMetricHistoryResponse,
 )
-from truefoundry.ml.autogen.client.models.list_metric_history_request_dto import (
+from truefoundry.ml._autogen.client.models.list_metric_history_request_dto import (
     ListMetricHistoryRequestDto,
 )
-from truefoundry.ml.autogen.client.models.list_metric_history_response_dto import (
+from truefoundry.ml._autogen.client.models.list_metric_history_response_dto import (
     ListMetricHistoryResponseDto,
 )
 from truefoundry.pydantic_v1 import StrictStr, validate_arguments