truefoundry 0.4.10__tar.gz → 0.5.0__tar.gz

{truefoundry-0.4.10 → truefoundry-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.10
+Version: 0.5.0
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.4.10 → truefoundry-0.5.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.4.10"
+version = "0.5.0"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"

{truefoundry-0.4.10 → truefoundry-0.5.0}/truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py

@@ -31,7 +31,7 @@ WORKDIR /app
 
 DOCKERFILE_TEMPLATE = Template(
     """
-FROM --platform=linux/amd64 ${python_image_repo}:${python_version}
+FROM ${python_image_repo}:${python_version}
 ENV PATH=/virtualenvs/venv/bin:$PATH
 RUN apt update && \
     DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends git && \
@@ -43,7 +43,7 @@ RUN apt update && \
 
 CUDA_DOCKERFILE_TEMPLATE = Template(
     """
-FROM --platform=linux/amd64 nvidia/cuda:${nvidia_cuda_image_tag}
+FROM nvidia/cuda:${nvidia_cuda_image_tag}
 ENV PATH=/virtualenvs/venv/bin:$PATH
 RUN echo "deb https://ppa.launchpadcontent.net/deadsnakes/ppa/ubuntu $(cat /etc/os-release | grep UBUNTU_CODENAME | cut -d = -f 2) main" >> /etc/apt/sources.list && \
     echo "deb-src https://ppa.launchpadcontent.net/deadsnakes/ppa/ubuntu $(cat /etc/os-release | grep UBUNTU_CODENAME | cut -d = -f 2) main" >> /etc/apt/sources.list && \

{truefoundry-0.4.10 → truefoundry-0.5.0}/truefoundry/deploy/lib/dao/application.py

@@ -179,10 +179,11 @@ def trigger_job(
         command=command_str if command_str else None,
         params=params if params else None,
     )
+    jobRunName = result.jobRunName
     previous_runs_url = f"{client.base_url.strip('/')}/deployments/{application_info.id}?tab=previousRuns"
     logger.info(
         f"{message}.\n"
-        f"You can check the status of your job run at {previous_runs_url}"
+        f"You can check the status of your job run at {previous_runs_url} with jobRunName: {jobRunName}"
     )
     return result
 

truefoundry-0.5.0/truefoundry/ml/__init__.py

@@ -0,0 +1,77 @@
+from truefoundry.ml.autogen.client.models import LibraryName, ModelVersionEnvironment
+from truefoundry.ml.enums import (
+    DataSlice,
+    FileFormat,
+    ModelFramework,
+    ModelType,
+    ViewType,
+)
+from truefoundry.ml.exceptions import MlFoundryException
+from truefoundry.ml.log_types import Image, Plot
+from truefoundry.ml.log_types.artifacts.artifact import (
+    ArtifactPath,
+    ArtifactVersion,
+    BlobStorageDirectory,
+)
+from truefoundry.ml.log_types.artifacts.dataset import DataDirectory, DataDirectoryPath
+from truefoundry.ml.log_types.artifacts.model import (
+    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",
+    "BlobStorageDirectory",
+    "DataDirectory",
+    "DataDirectoryPath",
+    "DataSlice",
+    "FileFormat",
+    "Image",
+    "MlFoundryRun",
+    "MlFoundryException",
+    "ModelVersionEnvironment",
+    "ModelFramework",
+    "ModelType",
+    "ModelVersion",
+    "Plot",
+    "ViewType",
+    "get_client",
+    "FastAIFramework",
+    "GluonFramework",
+    "H2OFramework",
+    "KerasFramework",
+    "LightGBMFramework",
+    "ONNXFramework",
+    "PaddleFramework",
+    "PyTorchFramework",
+    "SklearnFramework",
+    "SpaCyFramework",
+    "StatsModelsFramework",
+    "TensorFlowFramework",
+    "TransformersFramework",
+    "XGBoostFramework",
+    "LibraryName",
+    "ModelFrameworkType",
+]
+
+init_logger()

{truefoundry-0.4.10 → truefoundry-0.5.0}/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
 
@@ -61,6 +61,9 @@ from truefoundry.ml.autogen.client.models.artifact_response_dto import (
 )
 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_manifest import (
+    ArtifactVersionManifest,
+)
 from truefoundry.ml.autogen.client.models.artifact_version_response_dto import (
     ArtifactVersionResponseDto,
 )
@@ -88,6 +91,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 +143,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 (
@@ -165,11 +172,10 @@ 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.external_model_source import (
-    ExternalModelSource,
+from truefoundry.ml.autogen.client.models.external_artifact_source import (
+    ExternalArtifactSource,
 )
-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.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,
@@ -211,14 +217,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,
 )
@@ -291,6 +301,7 @@ from truefoundry.ml.autogen.client.models.log_metric_request_dto import (
 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.method import Method
 from truefoundry.ml.autogen.client.models.metric_collection_dto import (
     MetricCollectionDto,
@@ -300,8 +311,10 @@ 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_response_dto import ModelResponseDto
-from truefoundry.ml.autogen.client.models.model_schema_dto import ModelSchemaDto
 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_manifest import (
     ModelVersionManifest,
 )
@@ -320,10 +333,12 @@ 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,
 )
@@ -343,12 +358,21 @@ from truefoundry.ml.autogen.client.models.search_runs_request_dto import (
 from truefoundry.ml.autogen.client.models.search_runs_response_dto import (
     SearchRunsResponseDto,
 )
+from truefoundry.ml.autogen.client.models.serialization_format import (
+    SerializationFormat,
+)
 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.source import Source
+from truefoundry.ml.autogen.client.models.source1 import Source1
+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,16 +380,22 @@ 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.truefoundry_model_source import (
-    TruefoundryModelSource,
+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_artifact_source import (
+    TrueFoundryArtifactSource,
 )
 from truefoundry.ml.autogen.client.models.update_artifact_version_request_dto import (
     UpdateArtifactVersionRequestDto,
@@ -385,10 +415,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-0.4.10 → truefoundry-0.5.0}/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