truefoundry 0.4.6__py3-none-any.whl → 0.5.0rc2__py3-none-any.whl

truefoundry/ml/__init__.py

@@ -1,3 +1,4 @@
+from truefoundry.ml.autogen.client.models.library_name import LibraryName
 from truefoundry.ml.enums import (
     DataSlice,
     FileFormat,
@@ -10,15 +11,34 @@ from truefoundry.ml.log_types import Image, Plot
 from truefoundry.ml.log_types.artifacts.artifact import ArtifactPath, ArtifactVersion
 from truefoundry.ml.log_types.artifacts.dataset import DataDirectory, DataDirectoryPath
 from truefoundry.ml.log_types.artifacts.model import (
+    BlobStorageModelDirectory,
     ModelVersion,
 )
 from truefoundry.ml.logger import init_logger
 from truefoundry.ml.mlfoundry_api import get_client
 from truefoundry.ml.mlfoundry_run import MlFoundryRun
+from truefoundry.ml.model_framework import (
+    FastAIFramework,
+    GluonFramework,
+    H2OFramework,
+    KerasFramework,
+    LightGBMFramework,
+    ModelFrameworkType,
+    ONNXFramework,
+    PaddleFramework,
+    PyTorchFramework,
+    SklearnFramework,
+    SpaCyFramework,
+    StatsModelsFramework,
+    TensorFlowFramework,
+    TransformersFramework,
+    XGBoostFramework,
+)
 
 __all__ = [
     "ArtifactPath",
     "ArtifactVersion",
+    "BlobStorageModelDirectory",
     "DataDirectory",
     "DataDirectoryPath",
     "DataSlice",
@@ -32,6 +52,22 @@ __all__ = [
     "Plot",
     "ViewType",
     "get_client",
+    "FastAIFramework",
+    "GluonFramework",
+    "H2OFramework",
+    "KerasFramework",
+    "LightGBMFramework",
+    "ONNXFramework",
+    "PaddleFramework",
+    "PyTorchFramework",
+    "SklearnFramework",
+    "SpaCyFramework",
+    "StatsModelsFramework",
+    "TensorFlowFramework",
+    "TransformersFramework",
+    "XGBoostFramework",
+    "LibraryName",
+    "ModelFrameworkType",
 ]
 
 init_logger()

truefoundry/ml/autogen/client/__init__.py

@@ -19,14 +19,14 @@ __version__ = "0.1.0"
 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.python_deployment_config_api import (
-    PythonDeploymentConfigApi,
-)
 from truefoundry.ml.autogen.client.api.run_artifacts_api import RunArtifactsApi
 from truefoundry.ml.autogen.client.api.runs_api import RunsApi
 
@@ -88,6 +88,7 @@ from truefoundry.ml.autogen.client.models.chat_prompt_messages_inner import (
     ChatPromptMessagesInner,
 )
 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
@@ -139,6 +140,9 @@ from truefoundry.ml.autogen.client.models.create_run_request_dto import (
 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.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 (
@@ -168,6 +172,7 @@ from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTa
 from truefoundry.ml.autogen.client.models.external_model_source import (
     ExternalModelSource,
 )
+from truefoundry.ml.autogen.client.models.fast_ai_framework import FastAIFramework
 from truefoundry.ml.autogen.client.models.feature_dto import FeatureDto
 from truefoundry.ml.autogen.client.models.feature_value_type import FeatureValueType
 from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
@@ -211,14 +216,18 @@ from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_write_resp
 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.internal_metadata import InternalMetadata
+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,
 )
@@ -320,10 +329,13 @@ from truefoundry.ml.autogen.client.models.multi_part_upload_storage_provider imp
 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.prediction_type import PredictionType
+from truefoundry.ml.autogen.client.models.py_torch_framework import PyTorchFramework
 from truefoundry.ml.autogen.client.models.resolve_agent_app_response_dto import (
     ResolveAgentAppResponseDto,
 )
@@ -348,7 +360,12 @@ from truefoundry.ml.autogen.client.models.set_experiment_tag_request_dto import
 )
 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.source import Source
+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,
@@ -356,14 +373,20 @@ from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
 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.tensorflow_framework import (
-    TensorflowFramework,
+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.truefoundry_model_source import (
     TruefoundryModelSource,
 )
@@ -385,10 +408,10 @@ from truefoundry.ml.autogen.client.models.update_run_request_dto import (
 from truefoundry.ml.autogen.client.models.update_run_response_dto import (
     UpdateRunResponseDto,
 )
-from truefoundry.ml.autogen.client.models.upload_model_source import UploadModelSource
 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.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

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

@@ -4,13 +4,13 @@
 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.python_deployment_config_api import (
-    PythonDeploymentConfigApi,
-)
 from truefoundry.ml.autogen.client.api.run_artifacts_api import RunArtifactsApi
 from truefoundry.ml.autogen.client.api.runs_api import RunsApi

truefoundry/ml/autogen/client/api/deprecated_api.py

@@ -52,7 +52,7 @@ class DeprecatedApi:
         self.api_client = api_client
 
     @validate_arguments
-    def get_run_by_name_get(
+    def get_run_by_name_get_0(
         self,
         run_name: StrictStr,
         experiment_id: Optional[StrictStr] = None,
@@ -65,7 +65,7 @@ class DeprecatedApi:
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.get_run_by_name_get(run_name, experiment_id, experiment_name, async_req=True)
+        >>> thread = api.get_run_by_name_get_0(run_name, experiment_id, experiment_name, async_req=True)
         >>> result = thread.get()
 
         :param run_name: (required)
@@ -87,14 +87,14 @@ class DeprecatedApi:
         """
         kwargs["_return_http_data_only"] = True
         if "_preload_content" in kwargs:
-            message = "Error! Please call the get_run_by_name_get_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
+            message = "Error! Please call the get_run_by_name_get_0_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
             raise ValueError(message)
-        return self.get_run_by_name_get_with_http_info(
+        return self.get_run_by_name_get_0_with_http_info(
             run_name, experiment_id, experiment_name, **kwargs
         )  # noqa: E501
 
     @validate_arguments
-    def get_run_by_name_get_with_http_info(
+    def get_run_by_name_get_0_with_http_info(
         self,
         run_name: StrictStr,
         experiment_id: Optional[StrictStr] = None,
@@ -107,7 +107,7 @@ class DeprecatedApi:
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.get_run_by_name_get_with_http_info(run_name, experiment_id, experiment_name, async_req=True)
+        >>> thread = api.get_run_by_name_get_0_with_http_info(run_name, experiment_id, experiment_name, async_req=True)
         >>> result = thread.get()
 
         :param run_name: (required)
@@ -165,7 +165,7 @@ class DeprecatedApi:
             if _key not in _all_params:
                 raise ApiTypeError(
                     "Got an unexpected keyword argument '%s'"
-                    " to method get_run_by_name_get" % _key
+                    " to method get_run_by_name_get_0" % _key
                 )
             _params[_key] = _val
         del _params["kwargs"]