truefoundry 0.4.5__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.

Files changed (40) hide show
  1. truefoundry/ml/__init__.py +36 -0
  2. truefoundry/ml/autogen/client/__init__.py +29 -6
  3. truefoundry/ml/autogen/client/api/__init__.py +3 -3
  4. truefoundry/ml/autogen/client/api/deprecated_api.py +7 -7
  5. truefoundry/ml/autogen/client/api/generate_code_snippet_api.py +526 -0
  6. truefoundry/ml/autogen/client/models/__init__.py +26 -3
  7. truefoundry/ml/autogen/client/models/command.py +152 -0
  8. truefoundry/ml/autogen/client/models/create_workflow_task_config_request_dto.py +72 -0
  9. truefoundry/ml/autogen/client/models/external_model_source.py +3 -2
  10. truefoundry/ml/autogen/client/models/fast_ai_framework.py +75 -0
  11. truefoundry/ml/autogen/client/models/framework.py +250 -14
  12. truefoundry/ml/autogen/client/models/gluon_framework.py +74 -0
  13. truefoundry/ml/autogen/client/models/{upload_model_source.py → h2_o_framework.py} +11 -11
  14. truefoundry/ml/autogen/client/models/keras_framework.py +74 -0
  15. truefoundry/ml/autogen/client/models/light_gbm_framework.py +75 -0
  16. truefoundry/ml/autogen/client/models/model_version_manifest.py +1 -1
  17. truefoundry/ml/autogen/client/models/onnx_framework.py +74 -0
  18. truefoundry/ml/autogen/client/models/paddle_framework.py +75 -0
  19. truefoundry/ml/autogen/client/models/py_torch_framework.py +75 -0
  20. truefoundry/ml/autogen/client/models/sklearn_framework.py +75 -0
  21. truefoundry/ml/autogen/client/models/source.py +9 -32
  22. truefoundry/ml/autogen/client/models/spa_cy_framework.py +74 -0
  23. truefoundry/ml/autogen/client/models/stats_models_framework.py +75 -0
  24. truefoundry/ml/autogen/client/models/{tensorflow_framework.py → tensor_flow_framework.py} +10 -9
  25. truefoundry/ml/autogen/client/models/transformers_framework.py +3 -2
  26. truefoundry/ml/autogen/client/models/trigger_job_run_config_request_dto.py +90 -0
  27. truefoundry/ml/autogen/client/models/trigger_job_run_config_response_dto.py +71 -0
  28. truefoundry/ml/autogen/client/models/truefoundry_model_source.py +5 -3
  29. truefoundry/ml/autogen/client/models/xg_boost_framework.py +75 -0
  30. truefoundry/ml/autogen/client_README.md +22 -5
  31. truefoundry/ml/autogen/entities/artifacts.py +19 -2
  32. truefoundry/ml/log_types/artifacts/model.py +167 -177
  33. truefoundry/ml/mlfoundry_api.py +47 -18
  34. truefoundry/ml/mlfoundry_run.py +27 -12
  35. truefoundry/ml/model_framework.py +169 -0
  36. {truefoundry-0.4.5.dist-info → truefoundry-0.5.0rc1.dist-info}/METADATA +1 -1
  37. {truefoundry-0.4.5.dist-info → truefoundry-0.5.0rc1.dist-info}/RECORD +39 -23
  38. truefoundry/ml/autogen/client/api/python_deployment_config_api.py +0 -201
  39. {truefoundry-0.4.5.dist-info → truefoundry-0.5.0rc1.dist-info}/WHEEL +0 -0
  40. {truefoundry-0.4.5.dist-info → truefoundry-0.5.0rc1.dist-info}/entry_points.txt +0 -0
@@ -1,201 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- FastAPI
5
-
6
- No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
-
8
- The version of the OpenAPI document: 0.1.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
- import re # noqa: F401
15
-
16
- from truefoundry.ml.autogen.client.api_client import ApiClient
17
- from truefoundry.ml.autogen.client.api_response import ApiResponse
18
- from truefoundry.ml.autogen.client.exceptions import ( # noqa: F401
19
- ApiTypeError,
20
- ApiValueError,
21
- )
22
- from truefoundry.ml.autogen.client.models.create_python_deployment_config_request_dto import (
23
- CreatePythonDeploymentConfigRequestDto,
24
- )
25
- from truefoundry.ml.autogen.client.models.create_python_deployment_config_response_dto import (
26
- CreatePythonDeploymentConfigResponseDto,
27
- )
28
- from truefoundry.pydantic_v1 import validate_arguments
29
-
30
-
31
- class PythonDeploymentConfigApi:
32
- """NOTE: This class is auto generated by OpenAPI Generator
33
- Ref: https://openapi-generator.tech
34
-
35
- Do not edit the class manually.
36
- """
37
-
38
- def __init__(self, api_client=None) -> None:
39
- if api_client is None:
40
- api_client = ApiClient.get_default()
41
- self.api_client = api_client
42
-
43
- @validate_arguments
44
- def generate_py_development_config_post(
45
- self,
46
- create_python_deployment_config_request_dto: CreatePythonDeploymentConfigRequestDto,
47
- **kwargs,
48
- ) -> CreatePythonDeploymentConfigResponseDto: # noqa: E501
49
- """Generate Py Development Config # noqa: E501
50
-
51
- Generate Python Deployment Config # noqa: E501
52
- This method makes a synchronous HTTP request by default. To make an
53
- asynchronous HTTP request, please pass async_req=True
54
-
55
- >>> thread = api.generate_py_development_config_post(create_python_deployment_config_request_dto, async_req=True)
56
- >>> result = thread.get()
57
-
58
- :param create_python_deployment_config_request_dto: (required)
59
- :type create_python_deployment_config_request_dto: CreatePythonDeploymentConfigRequestDto
60
- :param async_req: Whether to execute the request asynchronously.
61
- :type async_req: bool, optional
62
- :param _request_timeout: timeout setting for this request.
63
- If one number provided, it will be total request
64
- timeout. It can also be a pair (tuple) of
65
- (connection, read) timeouts.
66
- :return: Returns the result object.
67
- If the method is called asynchronously,
68
- returns the request thread.
69
- :rtype: CreatePythonDeploymentConfigResponseDto
70
- """
71
- kwargs["_return_http_data_only"] = True
72
- if "_preload_content" in kwargs:
73
- message = "Error! Please call the generate_py_development_config_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data" # noqa: E501
74
- raise ValueError(message)
75
- return self.generate_py_development_config_post_with_http_info(
76
- create_python_deployment_config_request_dto, **kwargs
77
- ) # noqa: E501
78
-
79
- @validate_arguments
80
- def generate_py_development_config_post_with_http_info(
81
- self,
82
- create_python_deployment_config_request_dto: CreatePythonDeploymentConfigRequestDto,
83
- **kwargs,
84
- ) -> ApiResponse: # noqa: E501
85
- """Generate Py Development Config # noqa: E501
86
-
87
- Generate Python Deployment Config # noqa: E501
88
- This method makes a synchronous HTTP request by default. To make an
89
- asynchronous HTTP request, please pass async_req=True
90
-
91
- >>> thread = api.generate_py_development_config_post_with_http_info(create_python_deployment_config_request_dto, async_req=True)
92
- >>> result = thread.get()
93
-
94
- :param create_python_deployment_config_request_dto: (required)
95
- :type create_python_deployment_config_request_dto: CreatePythonDeploymentConfigRequestDto
96
- :param async_req: Whether to execute the request asynchronously.
97
- :type async_req: bool, optional
98
- :param _preload_content: if False, the ApiResponse.data will
99
- be set to none and raw_data will store the
100
- HTTP response body without reading/decoding.
101
- Default is True.
102
- :type _preload_content: bool, optional
103
- :param _return_http_data_only: response data instead of ApiResponse
104
- object with status code, headers, etc
105
- :type _return_http_data_only: bool, optional
106
- :param _request_timeout: timeout setting for this request. If one
107
- number provided, it will be total request
108
- timeout. It can also be a pair (tuple) of
109
- (connection, read) timeouts.
110
- :param _request_auth: set to override the auth_settings for an a single
111
- request; this effectively ignores the authentication
112
- in the spec for a single request.
113
- :type _request_auth: dict, optional
114
- :type _content_type: string, optional: force content-type for the request
115
- :return: Returns the result object.
116
- If the method is called asynchronously,
117
- returns the request thread.
118
- :rtype: tuple(CreatePythonDeploymentConfigResponseDto, status_code(int), headers(HTTPHeaderDict))
119
- """
120
-
121
- _params = locals()
122
-
123
- _all_params = ["create_python_deployment_config_request_dto"]
124
- _all_params.extend(
125
- [
126
- "async_req",
127
- "_return_http_data_only",
128
- "_preload_content",
129
- "_request_timeout",
130
- "_request_auth",
131
- "_content_type",
132
- "_headers",
133
- ]
134
- )
135
-
136
- # validate the arguments
137
- for _key, _val in _params["kwargs"].items():
138
- if _key not in _all_params:
139
- raise ApiTypeError(
140
- "Got an unexpected keyword argument '%s'"
141
- " to method generate_py_development_config_post" % _key
142
- )
143
- _params[_key] = _val
144
- del _params["kwargs"]
145
-
146
- _collection_formats = {}
147
-
148
- # process the path parameters
149
- _path_params = {}
150
-
151
- # process the query parameters
152
- _query_params = []
153
- # process the header parameters
154
- _header_params = dict(_params.get("_headers", {}))
155
- # process the form parameters
156
- _form_params = []
157
- _files = {}
158
- # process the body parameter
159
- _body_params = None
160
- if _params["create_python_deployment_config_request_dto"] is not None:
161
- _body_params = _params["create_python_deployment_config_request_dto"]
162
-
163
- # set the HTTP header `Accept`
164
- _header_params["Accept"] = self.api_client.select_header_accept(
165
- ["application/json"]
166
- ) # noqa: E501
167
-
168
- # set the HTTP header `Content-Type`
169
- _content_types_list = _params.get(
170
- "_content_type",
171
- self.api_client.select_header_content_type(["application/json"]),
172
- )
173
- if _content_types_list:
174
- _header_params["Content-Type"] = _content_types_list
175
-
176
- # authentication setting
177
- _auth_settings = [] # noqa: E501
178
-
179
- _response_types_map = {
180
- "200": "CreatePythonDeploymentConfigResponseDto",
181
- "422": "HTTPValidationError",
182
- }
183
-
184
- return self.api_client.call_api(
185
- "/api/2.0/mlflow/python-deployment-config/generate",
186
- "POST",
187
- _path_params,
188
- _query_params,
189
- _header_params,
190
- body=_body_params,
191
- post_params=_form_params,
192
- files=_files,
193
- response_types_map=_response_types_map,
194
- auth_settings=_auth_settings,
195
- async_req=_params.get("async_req"),
196
- _return_http_data_only=_params.get("_return_http_data_only"), # noqa: E501
197
- _preload_content=_params.get("_preload_content", True),
198
- _request_timeout=_params.get("_request_timeout"),
199
- collection_formats=_collection_formats,
200
- _request_auth=_params.get("_request_auth"),
201
- )