truefoundry 0.4.4rc12__py3-none-any.whl → 0.5.0rc1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of truefoundry might be problematic. Click here for more details.
- truefoundry/common/constants.py +6 -1
- truefoundry/common/utils.py +0 -18
- truefoundry/logger.py +1 -0
- truefoundry/ml/__init__.py +36 -0
- truefoundry/ml/artifact/truefoundry_artifact_repo.py +433 -415
- truefoundry/ml/autogen/client/__init__.py +29 -6
- truefoundry/ml/autogen/client/api/__init__.py +3 -3
- truefoundry/ml/autogen/client/api/deprecated_api.py +7 -7
- truefoundry/ml/autogen/client/api/generate_code_snippet_api.py +526 -0
- truefoundry/ml/autogen/client/models/__init__.py +26 -3
- truefoundry/ml/autogen/client/models/command.py +152 -0
- truefoundry/ml/autogen/client/models/create_workflow_task_config_request_dto.py +72 -0
- truefoundry/ml/autogen/client/models/external_model_source.py +3 -2
- truefoundry/ml/autogen/client/models/fast_ai_framework.py +75 -0
- truefoundry/ml/autogen/client/models/framework.py +250 -14
- truefoundry/ml/autogen/client/models/gluon_framework.py +74 -0
- truefoundry/ml/autogen/client/models/{upload_model_source.py → h2_o_framework.py} +11 -11
- truefoundry/ml/autogen/client/models/keras_framework.py +74 -0
- truefoundry/ml/autogen/client/models/light_gbm_framework.py +75 -0
- truefoundry/ml/autogen/client/models/model_version_manifest.py +1 -1
- truefoundry/ml/autogen/client/models/onnx_framework.py +74 -0
- truefoundry/ml/autogen/client/models/paddle_framework.py +75 -0
- truefoundry/ml/autogen/client/models/py_torch_framework.py +75 -0
- truefoundry/ml/autogen/client/models/sklearn_framework.py +75 -0
- truefoundry/ml/autogen/client/models/source.py +9 -32
- truefoundry/ml/autogen/client/models/spa_cy_framework.py +74 -0
- truefoundry/ml/autogen/client/models/stats_models_framework.py +75 -0
- truefoundry/ml/autogen/client/models/{tensorflow_framework.py → tensor_flow_framework.py} +10 -9
- truefoundry/ml/autogen/client/models/transformers_framework.py +3 -2
- truefoundry/ml/autogen/client/models/trigger_job_run_config_request_dto.py +90 -0
- truefoundry/ml/autogen/client/models/trigger_job_run_config_response_dto.py +71 -0
- truefoundry/ml/autogen/client/models/truefoundry_model_source.py +5 -3
- truefoundry/ml/autogen/client/models/xg_boost_framework.py +75 -0
- truefoundry/ml/autogen/client_README.md +22 -5
- truefoundry/ml/autogen/entities/artifacts.py +19 -2
- truefoundry/ml/log_types/artifacts/artifact.py +10 -6
- truefoundry/ml/log_types/artifacts/dataset.py +13 -5
- truefoundry/ml/log_types/artifacts/general_artifact.py +3 -1
- truefoundry/ml/log_types/artifacts/model.py +172 -194
- truefoundry/ml/log_types/artifacts/utils.py +43 -26
- truefoundry/ml/log_types/image/image.py +2 -0
- truefoundry/ml/log_types/plot.py +2 -0
- truefoundry/ml/mlfoundry_api.py +47 -18
- truefoundry/ml/mlfoundry_run.py +27 -12
- truefoundry/ml/model_framework.py +169 -0
- truefoundry/workflow/__init__.py +3 -1
- truefoundry/workflow/remote_filesystem/__init__.py +8 -0
- truefoundry/workflow/remote_filesystem/logger.py +36 -0
- truefoundry/{common → workflow/remote_filesystem}/tfy_signed_url_client.py +1 -2
- truefoundry/{common → workflow/remote_filesystem}/tfy_signed_url_fs.py +5 -2
- {truefoundry-0.4.4rc12.dist-info → truefoundry-0.5.0rc1.dist-info}/METADATA +1 -1
- {truefoundry-0.4.4rc12.dist-info → truefoundry-0.5.0rc1.dist-info}/RECORD +54 -36
- truefoundry/ml/autogen/client/api/python_deployment_config_api.py +0 -201
- {truefoundry-0.4.4rc12.dist-info → truefoundry-0.5.0rc1.dist-info}/WHEEL +0 -0
- {truefoundry-0.4.4rc12.dist-info → truefoundry-0.5.0rc1.dist-info}/entry_points.txt +0 -0
|
@@ -19,14 +19,14 @@ __version__ = "0.1.0"
|
|
|
19
19
|
from truefoundry.ml.autogen.client.api.auth_api import AuthApi
|
|
20
20
|
from truefoundry.ml.autogen.client.api.deprecated_api import DeprecatedApi
|
|
21
21
|
from truefoundry.ml.autogen.client.api.experiments_api import ExperimentsApi
|
|
22
|
+
from truefoundry.ml.autogen.client.api.generate_code_snippet_api import (
|
|
23
|
+
GenerateCodeSnippetApi,
|
|
24
|
+
)
|
|
22
25
|
from truefoundry.ml.autogen.client.api.health_api import HealthApi
|
|
23
26
|
from truefoundry.ml.autogen.client.api.metrics_api import MetricsApi
|
|
24
27
|
from truefoundry.ml.autogen.client.api.mlfoundry_artifacts_api import (
|
|
25
28
|
MlfoundryArtifactsApi,
|
|
26
29
|
)
|
|
27
|
-
from truefoundry.ml.autogen.client.api.python_deployment_config_api import (
|
|
28
|
-
PythonDeploymentConfigApi,
|
|
29
|
-
)
|
|
30
30
|
from truefoundry.ml.autogen.client.api.run_artifacts_api import RunArtifactsApi
|
|
31
31
|
from truefoundry.ml.autogen.client.api.runs_api import RunsApi
|
|
32
32
|
|
|
@@ -88,6 +88,7 @@ from truefoundry.ml.autogen.client.models.chat_prompt_messages_inner import (
|
|
|
88
88
|
ChatPromptMessagesInner,
|
|
89
89
|
)
|
|
90
90
|
from truefoundry.ml.autogen.client.models.columns_dto import ColumnsDto
|
|
91
|
+
from truefoundry.ml.autogen.client.models.command import Command
|
|
91
92
|
from truefoundry.ml.autogen.client.models.content import Content
|
|
92
93
|
from truefoundry.ml.autogen.client.models.content1 import Content1
|
|
93
94
|
from truefoundry.ml.autogen.client.models.content2 import Content2
|
|
@@ -139,6 +140,9 @@ from truefoundry.ml.autogen.client.models.create_run_request_dto import (
|
|
|
139
140
|
from truefoundry.ml.autogen.client.models.create_run_response_dto import (
|
|
140
141
|
CreateRunResponseDto,
|
|
141
142
|
)
|
|
143
|
+
from truefoundry.ml.autogen.client.models.create_workflow_task_config_request_dto import (
|
|
144
|
+
CreateWorkflowTaskConfigRequestDto,
|
|
145
|
+
)
|
|
142
146
|
from truefoundry.ml.autogen.client.models.dataset_dto import DatasetDto
|
|
143
147
|
from truefoundry.ml.autogen.client.models.dataset_response_dto import DatasetResponseDto
|
|
144
148
|
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
|
|
|
168
172
|
from truefoundry.ml.autogen.client.models.external_model_source import (
|
|
169
173
|
ExternalModelSource,
|
|
170
174
|
)
|
|
175
|
+
from truefoundry.ml.autogen.client.models.fast_ai_framework import FastAIFramework
|
|
171
176
|
from truefoundry.ml.autogen.client.models.feature_dto import FeatureDto
|
|
172
177
|
from truefoundry.ml.autogen.client.models.feature_value_type import FeatureValueType
|
|
173
178
|
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
|
|
|
211
216
|
from truefoundry.ml.autogen.client.models.get_tenant_id_response_dto import (
|
|
212
217
|
GetTenantIdResponseDto,
|
|
213
218
|
)
|
|
219
|
+
from truefoundry.ml.autogen.client.models.gluon_framework import GluonFramework
|
|
220
|
+
from truefoundry.ml.autogen.client.models.h2_o_framework import H2OFramework
|
|
214
221
|
from truefoundry.ml.autogen.client.models.http_validation_error import (
|
|
215
222
|
HTTPValidationError,
|
|
216
223
|
)
|
|
217
224
|
from truefoundry.ml.autogen.client.models.image_content_part import ImageContentPart
|
|
218
225
|
from truefoundry.ml.autogen.client.models.image_url import ImageUrl
|
|
219
226
|
from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
|
|
227
|
+
from truefoundry.ml.autogen.client.models.keras_framework import KerasFramework
|
|
220
228
|
from truefoundry.ml.autogen.client.models.latest_run_log_dto import LatestRunLogDto
|
|
221
229
|
from truefoundry.ml.autogen.client.models.library_name import LibraryName
|
|
230
|
+
from truefoundry.ml.autogen.client.models.light_gbm_framework import LightGBMFramework
|
|
222
231
|
from truefoundry.ml.autogen.client.models.list_artifact_versions_request_dto import (
|
|
223
232
|
ListArtifactVersionsRequestDto,
|
|
224
233
|
)
|
|
@@ -320,10 +329,13 @@ from truefoundry.ml.autogen.client.models.multi_part_upload_storage_provider imp
|
|
|
320
329
|
from truefoundry.ml.autogen.client.models.notify_artifact_version_failure_dto import (
|
|
321
330
|
NotifyArtifactVersionFailureDto,
|
|
322
331
|
)
|
|
332
|
+
from truefoundry.ml.autogen.client.models.onnx_framework import ONNXFramework
|
|
323
333
|
from truefoundry.ml.autogen.client.models.openapi_spec import OpenapiSpec
|
|
334
|
+
from truefoundry.ml.autogen.client.models.paddle_framework import PaddleFramework
|
|
324
335
|
from truefoundry.ml.autogen.client.models.param_dto import ParamDto
|
|
325
336
|
from truefoundry.ml.autogen.client.models.parameters import Parameters
|
|
326
337
|
from truefoundry.ml.autogen.client.models.prediction_type import PredictionType
|
|
338
|
+
from truefoundry.ml.autogen.client.models.py_torch_framework import PyTorchFramework
|
|
327
339
|
from truefoundry.ml.autogen.client.models.resolve_agent_app_response_dto import (
|
|
328
340
|
ResolveAgentAppResponseDto,
|
|
329
341
|
)
|
|
@@ -348,7 +360,12 @@ from truefoundry.ml.autogen.client.models.set_experiment_tag_request_dto import
|
|
|
348
360
|
)
|
|
349
361
|
from truefoundry.ml.autogen.client.models.set_tag_request_dto import SetTagRequestDto
|
|
350
362
|
from truefoundry.ml.autogen.client.models.signed_url_dto import SignedURLDto
|
|
363
|
+
from truefoundry.ml.autogen.client.models.sklearn_framework import SklearnFramework
|
|
351
364
|
from truefoundry.ml.autogen.client.models.source import Source
|
|
365
|
+
from truefoundry.ml.autogen.client.models.spa_cy_framework import SpaCyFramework
|
|
366
|
+
from truefoundry.ml.autogen.client.models.stats_models_framework import (
|
|
367
|
+
StatsModelsFramework,
|
|
368
|
+
)
|
|
352
369
|
from truefoundry.ml.autogen.client.models.stop import Stop
|
|
353
370
|
from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
|
|
354
371
|
StoreRunLogsRequestDto,
|
|
@@ -356,14 +373,20 @@ from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
|
|
|
356
373
|
from truefoundry.ml.autogen.client.models.subject import Subject
|
|
357
374
|
from truefoundry.ml.autogen.client.models.subject_type import SubjectType
|
|
358
375
|
from truefoundry.ml.autogen.client.models.system_message import SystemMessage
|
|
359
|
-
from truefoundry.ml.autogen.client.models.
|
|
360
|
-
|
|
376
|
+
from truefoundry.ml.autogen.client.models.tensor_flow_framework import (
|
|
377
|
+
TensorFlowFramework,
|
|
361
378
|
)
|
|
362
379
|
from truefoundry.ml.autogen.client.models.text import Text
|
|
363
380
|
from truefoundry.ml.autogen.client.models.text_content_part import TextContentPart
|
|
364
381
|
from truefoundry.ml.autogen.client.models.transformers_framework import (
|
|
365
382
|
TransformersFramework,
|
|
366
383
|
)
|
|
384
|
+
from truefoundry.ml.autogen.client.models.trigger_job_run_config_request_dto import (
|
|
385
|
+
TriggerJobRunConfigRequestDto,
|
|
386
|
+
)
|
|
387
|
+
from truefoundry.ml.autogen.client.models.trigger_job_run_config_response_dto import (
|
|
388
|
+
TriggerJobRunConfigResponseDto,
|
|
389
|
+
)
|
|
367
390
|
from truefoundry.ml.autogen.client.models.truefoundry_model_source import (
|
|
368
391
|
TruefoundryModelSource,
|
|
369
392
|
)
|
|
@@ -385,10 +408,10 @@ from truefoundry.ml.autogen.client.models.update_run_request_dto import (
|
|
|
385
408
|
from truefoundry.ml.autogen.client.models.update_run_response_dto import (
|
|
386
409
|
UpdateRunResponseDto,
|
|
387
410
|
)
|
|
388
|
-
from truefoundry.ml.autogen.client.models.upload_model_source import UploadModelSource
|
|
389
411
|
from truefoundry.ml.autogen.client.models.url import Url
|
|
390
412
|
from truefoundry.ml.autogen.client.models.user_message import UserMessage
|
|
391
413
|
from truefoundry.ml.autogen.client.models.validation_error import ValidationError
|
|
392
414
|
from truefoundry.ml.autogen.client.models.validation_error_loc_inner import (
|
|
393
415
|
ValidationErrorLocInner,
|
|
394
416
|
)
|
|
417
|
+
from truefoundry.ml.autogen.client.models.xg_boost_framework import XGBoostFramework
|
|
@@ -4,13 +4,13 @@
|
|
|
4
4
|
from truefoundry.ml.autogen.client.api.auth_api import AuthApi
|
|
5
5
|
from truefoundry.ml.autogen.client.api.deprecated_api import DeprecatedApi
|
|
6
6
|
from truefoundry.ml.autogen.client.api.experiments_api import ExperimentsApi
|
|
7
|
+
from truefoundry.ml.autogen.client.api.generate_code_snippet_api import (
|
|
8
|
+
GenerateCodeSnippetApi,
|
|
9
|
+
)
|
|
7
10
|
from truefoundry.ml.autogen.client.api.health_api import HealthApi
|
|
8
11
|
from truefoundry.ml.autogen.client.api.metrics_api import MetricsApi
|
|
9
12
|
from truefoundry.ml.autogen.client.api.mlfoundry_artifacts_api import (
|
|
10
13
|
MlfoundryArtifactsApi,
|
|
11
14
|
)
|
|
12
|
-
from truefoundry.ml.autogen.client.api.python_deployment_config_api import (
|
|
13
|
-
PythonDeploymentConfigApi,
|
|
14
|
-
)
|
|
15
15
|
from truefoundry.ml.autogen.client.api.run_artifacts_api import RunArtifactsApi
|
|
16
16
|
from truefoundry.ml.autogen.client.api.runs_api import RunsApi
|
|
@@ -52,7 +52,7 @@ class DeprecatedApi:
|
|
|
52
52
|
self.api_client = api_client
|
|
53
53
|
|
|
54
54
|
@validate_arguments
|
|
55
|
-
def
|
|
55
|
+
def get_run_by_name_get_0(
|
|
56
56
|
self,
|
|
57
57
|
run_name: StrictStr,
|
|
58
58
|
experiment_id: Optional[StrictStr] = None,
|
|
@@ -65,7 +65,7 @@ class DeprecatedApi:
|
|
|
65
65
|
This method makes a synchronous HTTP request by default. To make an
|
|
66
66
|
asynchronous HTTP request, please pass async_req=True
|
|
67
67
|
|
|
68
|
-
>>> thread = api.
|
|
68
|
+
>>> thread = api.get_run_by_name_get_0(run_name, experiment_id, experiment_name, async_req=True)
|
|
69
69
|
>>> result = thread.get()
|
|
70
70
|
|
|
71
71
|
:param run_name: (required)
|
|
@@ -87,14 +87,14 @@ class DeprecatedApi:
|
|
|
87
87
|
"""
|
|
88
88
|
kwargs["_return_http_data_only"] = True
|
|
89
89
|
if "_preload_content" in kwargs:
|
|
90
|
-
message = "Error! Please call the
|
|
90
|
+
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
|
|
91
91
|
raise ValueError(message)
|
|
92
|
-
return self.
|
|
92
|
+
return self.get_run_by_name_get_0_with_http_info(
|
|
93
93
|
run_name, experiment_id, experiment_name, **kwargs
|
|
94
94
|
) # noqa: E501
|
|
95
95
|
|
|
96
96
|
@validate_arguments
|
|
97
|
-
def
|
|
97
|
+
def get_run_by_name_get_0_with_http_info(
|
|
98
98
|
self,
|
|
99
99
|
run_name: StrictStr,
|
|
100
100
|
experiment_id: Optional[StrictStr] = None,
|
|
@@ -107,7 +107,7 @@ class DeprecatedApi:
|
|
|
107
107
|
This method makes a synchronous HTTP request by default. To make an
|
|
108
108
|
asynchronous HTTP request, please pass async_req=True
|
|
109
109
|
|
|
110
|
-
>>> thread = api.
|
|
110
|
+
>>> thread = api.get_run_by_name_get_0_with_http_info(run_name, experiment_id, experiment_name, async_req=True)
|
|
111
111
|
>>> result = thread.get()
|
|
112
112
|
|
|
113
113
|
:param run_name: (required)
|
|
@@ -165,7 +165,7 @@ class DeprecatedApi:
|
|
|
165
165
|
if _key not in _all_params:
|
|
166
166
|
raise ApiTypeError(
|
|
167
167
|
"Got an unexpected keyword argument '%s'"
|
|
168
|
-
" to method
|
|
168
|
+
" to method get_run_by_name_get_0" % _key
|
|
169
169
|
)
|
|
170
170
|
_params[_key] = _val
|
|
171
171
|
del _params["kwargs"]
|