lightning-sdk 0.1.38__py3-none-any.whl → 0.1.40__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.
Files changed (53) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/_mmt/__init__.py +3 -0
  3. lightning_sdk/_mmt/base.py +180 -0
  4. lightning_sdk/_mmt/mmt.py +161 -0
  5. lightning_sdk/_mmt/v1.py +69 -0
  6. lightning_sdk/_mmt/v2.py +141 -0
  7. lightning_sdk/api/deployment_api.py +0 -2
  8. lightning_sdk/api/job_api.py +4 -0
  9. lightning_sdk/api/mmt_api.py +147 -0
  10. lightning_sdk/api/teamspace_api.py +4 -11
  11. lightning_sdk/api/utils.py +6 -3
  12. lightning_sdk/cli/download.py +3 -5
  13. lightning_sdk/cli/mmt.py +137 -0
  14. lightning_sdk/cli/run.py +16 -0
  15. lightning_sdk/cli/upload.py +3 -10
  16. lightning_sdk/job/base.py +24 -3
  17. lightning_sdk/job/job.py +10 -1
  18. lightning_sdk/job/v1.py +7 -1
  19. lightning_sdk/job/v2.py +18 -9
  20. lightning_sdk/lightning_cloud/openapi/__init__.py +7 -3
  21. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +90 -284
  22. lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py +6 -1
  23. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +235 -1
  24. lightning_sdk/lightning_cloud/openapi/models/__init__.py +7 -3
  25. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/id_start_body.py +29 -3
  27. lightning_sdk/lightning_cloud/openapi/models/model_id_visibility_body.py +123 -0
  28. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  29. lightning_sdk/lightning_cloud/openapi/models/v1_aws_direct_v1.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/{project_id_agentmanagedmodels_body.py → v1_body.py} +21 -47
  31. lightning_sdk/lightning_cloud/openapi/models/v1_data_path.py +29 -3
  32. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py +201 -0
  34. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py +27 -1
  35. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +53 -53
  36. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +27 -1
  37. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model_abilities.py +175 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_model.py +29 -3
  39. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +53 -1
  40. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py +1 -2
  41. lightning_sdk/lightning_cloud/openapi/models/v1_query_param.py +175 -0
  42. lightning_sdk/lightning_cloud/openapi/models/{v1_list_managed_models_response.py → v1_resource_visibility.py} +23 -23
  43. lightning_sdk/lightning_cloud/openapi/models/{v1_delete_managed_model_response.py → v1_update_model_visibility_response.py} +6 -6
  44. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +27 -1
  45. lightning_sdk/models.py +153 -0
  46. lightning_sdk/teamspace.py +15 -11
  47. {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/METADATA +1 -1
  48. {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/RECORD +52 -41
  49. {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/entry_points.txt +1 -0
  50. lightning_sdk/cli/models.py +0 -68
  51. {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/LICENSE +0 -0
  52. {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/WHEEL +0 -0
  53. {lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.40.dist-info}/top_level.txt +0 -0
@@ -43,6 +43,7 @@ class V1AWSDirectV1(object):
43
43
  swagger_types = {
44
44
  'bucket_endpoint': 'str',
45
45
  'bucket_name': 'str',
46
+ 'cloud_formation_disabled': 'bool',
46
47
  'efs_enabled': 'bool',
47
48
  'external_id': 'str',
48
49
  'primary_region': 'str',
@@ -56,6 +57,7 @@ class V1AWSDirectV1(object):
56
57
  attribute_map = {
57
58
  'bucket_endpoint': 'bucketEndpoint',
58
59
  'bucket_name': 'bucketName',
60
+ 'cloud_formation_disabled': 'cloudFormationDisabled',
59
61
  'efs_enabled': 'efsEnabled',
60
62
  'external_id': 'externalId',
61
63
  'primary_region': 'primaryRegion',
@@ -66,10 +68,11 @@ class V1AWSDirectV1(object):
66
68
  'use_launch_templates': 'useLaunchTemplates'
67
69
  }
68
70
 
69
- def __init__(self, bucket_endpoint: 'str' =None, bucket_name: 'str' =None, efs_enabled: 'bool' =None, external_id: 'str' =None, primary_region: 'str' =None, regional_load_balancers: 'list[V1RegionalLoadBalancer]' =None, regions: 'list[str]' =None, role_arn: 'str' =None, source_cidr_ips: 'list[str]' =None, use_launch_templates: 'bool' =None): # noqa: E501
71
+ def __init__(self, bucket_endpoint: 'str' =None, bucket_name: 'str' =None, cloud_formation_disabled: 'bool' =None, efs_enabled: 'bool' =None, external_id: 'str' =None, primary_region: 'str' =None, regional_load_balancers: 'list[V1RegionalLoadBalancer]' =None, regions: 'list[str]' =None, role_arn: 'str' =None, source_cidr_ips: 'list[str]' =None, use_launch_templates: 'bool' =None): # noqa: E501
70
72
  """V1AWSDirectV1 - a model defined in Swagger""" # noqa: E501
71
73
  self._bucket_endpoint = None
72
74
  self._bucket_name = None
75
+ self._cloud_formation_disabled = None
73
76
  self._efs_enabled = None
74
77
  self._external_id = None
75
78
  self._primary_region = None
@@ -83,6 +86,8 @@ class V1AWSDirectV1(object):
83
86
  self.bucket_endpoint = bucket_endpoint
84
87
  if bucket_name is not None:
85
88
  self.bucket_name = bucket_name
89
+ if cloud_formation_disabled is not None:
90
+ self.cloud_formation_disabled = cloud_formation_disabled
86
91
  if efs_enabled is not None:
87
92
  self.efs_enabled = efs_enabled
88
93
  if external_id is not None:
@@ -142,6 +147,27 @@ class V1AWSDirectV1(object):
142
147
 
143
148
  self._bucket_name = bucket_name
144
149
 
150
+ @property
151
+ def cloud_formation_disabled(self) -> 'bool':
152
+ """Gets the cloud_formation_disabled of this V1AWSDirectV1. # noqa: E501
153
+
154
+
155
+ :return: The cloud_formation_disabled of this V1AWSDirectV1. # noqa: E501
156
+ :rtype: bool
157
+ """
158
+ return self._cloud_formation_disabled
159
+
160
+ @cloud_formation_disabled.setter
161
+ def cloud_formation_disabled(self, cloud_formation_disabled: 'bool'):
162
+ """Sets the cloud_formation_disabled of this V1AWSDirectV1.
163
+
164
+
165
+ :param cloud_formation_disabled: The cloud_formation_disabled of this V1AWSDirectV1. # noqa: E501
166
+ :type: bool
167
+ """
168
+
169
+ self._cloud_formation_disabled = cloud_formation_disabled
170
+
145
171
  @property
146
172
  def efs_enabled(self) -> 'bool':
147
173
  """Gets the efs_enabled of this V1AWSDirectV1. # noqa: E501
@@ -28,7 +28,7 @@ if TYPE_CHECKING:
28
28
  from datetime import datetime
29
29
  from lightning_sdk.lightning_cloud.openapi.models import *
30
30
 
31
- class ProjectIdAgentmanagedmodelsBody(object):
31
+ class V1Body(object):
32
32
  """NOTE: This class is auto generated by the swagger code generator program.
33
33
 
34
34
  Do not edit the class manually.
@@ -41,66 +41,40 @@ class ProjectIdAgentmanagedmodelsBody(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'model': 'V1ManagedModel',
45
- 'org_id': 'str'
44
+ 'raw_json': 'str'
46
45
  }
47
46
 
48
47
  attribute_map = {
49
- 'model': 'model',
50
- 'org_id': 'orgId'
48
+ 'raw_json': 'rawJson'
51
49
  }
52
50
 
53
- def __init__(self, model: 'V1ManagedModel' =None, org_id: 'str' =None): # noqa: E501
54
- """ProjectIdAgentmanagedmodelsBody - a model defined in Swagger""" # noqa: E501
55
- self._model = None
56
- self._org_id = None
51
+ def __init__(self, raw_json: 'str' =None): # noqa: E501
52
+ """V1Body - a model defined in Swagger""" # noqa: E501
53
+ self._raw_json = None
57
54
  self.discriminator = None
58
- if model is not None:
59
- self.model = model
60
- if org_id is not None:
61
- self.org_id = org_id
55
+ if raw_json is not None:
56
+ self.raw_json = raw_json
62
57
 
63
58
  @property
64
- def model(self) -> 'V1ManagedModel':
65
- """Gets the model of this ProjectIdAgentmanagedmodelsBody. # noqa: E501
59
+ def raw_json(self) -> 'str':
60
+ """Gets the raw_json of this V1Body. # noqa: E501
66
61
 
67
62
 
68
- :return: The model of this ProjectIdAgentmanagedmodelsBody. # noqa: E501
69
- :rtype: V1ManagedModel
70
- """
71
- return self._model
72
-
73
- @model.setter
74
- def model(self, model: 'V1ManagedModel'):
75
- """Sets the model of this ProjectIdAgentmanagedmodelsBody.
76
-
77
-
78
- :param model: The model of this ProjectIdAgentmanagedmodelsBody. # noqa: E501
79
- :type: V1ManagedModel
80
- """
81
-
82
- self._model = model
83
-
84
- @property
85
- def org_id(self) -> 'str':
86
- """Gets the org_id of this ProjectIdAgentmanagedmodelsBody. # noqa: E501
87
-
88
-
89
- :return: The org_id of this ProjectIdAgentmanagedmodelsBody. # noqa: E501
63
+ :return: The raw_json of this V1Body. # noqa: E501
90
64
  :rtype: str
91
65
  """
92
- return self._org_id
66
+ return self._raw_json
93
67
 
94
- @org_id.setter
95
- def org_id(self, org_id: 'str'):
96
- """Sets the org_id of this ProjectIdAgentmanagedmodelsBody.
68
+ @raw_json.setter
69
+ def raw_json(self, raw_json: 'str'):
70
+ """Sets the raw_json of this V1Body.
97
71
 
98
72
 
99
- :param org_id: The org_id of this ProjectIdAgentmanagedmodelsBody. # noqa: E501
73
+ :param raw_json: The raw_json of this V1Body. # noqa: E501
100
74
  :type: str
101
75
  """
102
76
 
103
- self._org_id = org_id
77
+ self._raw_json = raw_json
104
78
 
105
79
  def to_dict(self) -> dict:
106
80
  """Returns the model properties as a dict"""
@@ -123,7 +97,7 @@ class ProjectIdAgentmanagedmodelsBody(object):
123
97
  ))
124
98
  else:
125
99
  result[attr] = value
126
- if issubclass(ProjectIdAgentmanagedmodelsBody, dict):
100
+ if issubclass(V1Body, dict):
127
101
  for key, value in self.items():
128
102
  result[key] = value
129
103
 
@@ -137,13 +111,13 @@ class ProjectIdAgentmanagedmodelsBody(object):
137
111
  """For `print` and `pprint`"""
138
112
  return self.to_str()
139
113
 
140
- def __eq__(self, other: 'ProjectIdAgentmanagedmodelsBody') -> bool:
114
+ def __eq__(self, other: 'V1Body') -> bool:
141
115
  """Returns true if both objects are equal"""
142
- if not isinstance(other, ProjectIdAgentmanagedmodelsBody):
116
+ if not isinstance(other, V1Body):
143
117
  return False
144
118
 
145
119
  return self.__dict__ == other.__dict__
146
120
 
147
- def __ne__(self, other: 'ProjectIdAgentmanagedmodelsBody') -> bool:
121
+ def __ne__(self, other: 'V1Body') -> bool:
148
122
  """Returns true if both objects are not equal"""
149
123
  return not self == other
@@ -43,20 +43,23 @@ class V1DataPath(object):
43
43
  swagger_types = {
44
44
  'single_file': 'bool',
45
45
  'stored_path': 'str',
46
- 'target_path': 'str'
46
+ 'target_path': 'str',
47
+ 'updated_at': 'datetime'
47
48
  }
48
49
 
49
50
  attribute_map = {
50
51
  'single_file': 'singleFile',
51
52
  'stored_path': 'storedPath',
52
- 'target_path': 'targetPath'
53
+ 'target_path': 'targetPath',
54
+ 'updated_at': 'updatedAt'
53
55
  }
54
56
 
55
- def __init__(self, single_file: 'bool' =None, stored_path: 'str' =None, target_path: 'str' =None): # noqa: E501
57
+ def __init__(self, single_file: 'bool' =None, stored_path: 'str' =None, target_path: 'str' =None, updated_at: 'datetime' =None): # noqa: E501
56
58
  """V1DataPath - a model defined in Swagger""" # noqa: E501
57
59
  self._single_file = None
58
60
  self._stored_path = None
59
61
  self._target_path = None
62
+ self._updated_at = None
60
63
  self.discriminator = None
61
64
  if single_file is not None:
62
65
  self.single_file = single_file
@@ -64,6 +67,8 @@ class V1DataPath(object):
64
67
  self.stored_path = stored_path
65
68
  if target_path is not None:
66
69
  self.target_path = target_path
70
+ if updated_at is not None:
71
+ self.updated_at = updated_at
67
72
 
68
73
  @property
69
74
  def single_file(self) -> 'bool':
@@ -128,6 +133,27 @@ class V1DataPath(object):
128
133
 
129
134
  self._target_path = target_path
130
135
 
136
+ @property
137
+ def updated_at(self) -> 'datetime':
138
+ """Gets the updated_at of this V1DataPath. # noqa: E501
139
+
140
+
141
+ :return: The updated_at of this V1DataPath. # noqa: E501
142
+ :rtype: datetime
143
+ """
144
+ return self._updated_at
145
+
146
+ @updated_at.setter
147
+ def updated_at(self, updated_at: 'datetime'):
148
+ """Sets the updated_at of this V1DataPath.
149
+
150
+
151
+ :param updated_at: The updated_at of this V1DataPath. # noqa: E501
152
+ :type: datetime
153
+ """
154
+
155
+ self._updated_at = updated_at
156
+
131
157
  def to_dict(self) -> dict:
132
158
  """Returns the model properties as a dict"""
133
159
  result = {}
@@ -41,6 +41,7 @@ class V1Deployment(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'apis': 'list[V1DeploymentAPI]',
44
45
  'autoscaling': 'V1AutoscalingSpec',
45
46
  'cloudspace_id': 'str',
46
47
  'created_at': 'datetime',
@@ -62,6 +63,7 @@ class V1Deployment(object):
62
63
  }
63
64
 
64
65
  attribute_map = {
66
+ 'apis': 'apis',
65
67
  'autoscaling': 'autoscaling',
66
68
  'cloudspace_id': 'cloudspaceId',
67
69
  'created_at': 'createdAt',
@@ -82,8 +84,9 @@ class V1Deployment(object):
82
84
  'user_id': 'userId'
83
85
  }
84
86
 
85
- def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, project_id: 'str' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, template_id: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
87
+ def __init__(self, apis: 'list[V1DeploymentAPI]' =None, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, project_id: 'str' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, template_id: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
86
88
  """V1Deployment - a model defined in Swagger""" # noqa: E501
89
+ self._apis = None
87
90
  self._autoscaling = None
88
91
  self._cloudspace_id = None
89
92
  self._created_at = None
@@ -103,6 +106,8 @@ class V1Deployment(object):
103
106
  self._updated_at = None
104
107
  self._user_id = None
105
108
  self.discriminator = None
109
+ if apis is not None:
110
+ self.apis = apis
106
111
  if autoscaling is not None:
107
112
  self.autoscaling = autoscaling
108
113
  if cloudspace_id is not None:
@@ -140,6 +145,27 @@ class V1Deployment(object):
140
145
  if user_id is not None:
141
146
  self.user_id = user_id
142
147
 
148
+ @property
149
+ def apis(self) -> 'list[V1DeploymentAPI]':
150
+ """Gets the apis of this V1Deployment. # noqa: E501
151
+
152
+
153
+ :return: The apis of this V1Deployment. # noqa: E501
154
+ :rtype: list[V1DeploymentAPI]
155
+ """
156
+ return self._apis
157
+
158
+ @apis.setter
159
+ def apis(self, apis: 'list[V1DeploymentAPI]'):
160
+ """Sets the apis of this V1Deployment.
161
+
162
+
163
+ :param apis: The apis of this V1Deployment. # noqa: E501
164
+ :type: list[V1DeploymentAPI]
165
+ """
166
+
167
+ self._apis = apis
168
+
143
169
  @property
144
170
  def autoscaling(self) -> 'V1AutoscalingSpec':
145
171
  """Gets the autoscaling of this V1Deployment. # noqa: E501
@@ -0,0 +1,201 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ external/v1/auth_service.proto
5
+
6
+ No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7
+
8
+ OpenAPI spec version: version not set
9
+
10
+ Generated by: https://github.com/swagger-api/swagger-codegen.git
11
+
12
+ NOTE
13
+ ----
14
+ standard swagger-codegen-cli for this python client has been modified
15
+ by custom templates. The purpose of these templates is to include
16
+ typing information in the API and Model code. Please refer to the
17
+ main grid repository for more info
18
+ """
19
+
20
+ import pprint
21
+ import re # noqa: F401
22
+
23
+ from typing import TYPE_CHECKING
24
+
25
+ import six
26
+
27
+ if TYPE_CHECKING:
28
+ from datetime import datetime
29
+ from lightning_sdk.lightning_cloud.openapi.models import *
30
+
31
+ class V1DeploymentAPI(object):
32
+ """NOTE: This class is auto generated by the swagger code generator program.
33
+
34
+ Do not edit the class manually.
35
+ """
36
+ """
37
+ Attributes:
38
+ swagger_types (dict): The key is attribute name
39
+ and the value is attribute type.
40
+ attribute_map (dict): The key is attribute name
41
+ and the value is json key in definition.
42
+ """
43
+ swagger_types = {
44
+ 'body': 'V1Body',
45
+ 'method': 'str',
46
+ 'path': 'str',
47
+ '_query_params': 'list[V1QueryParam]'
48
+ }
49
+
50
+ attribute_map = {
51
+ 'body': 'body',
52
+ 'method': 'method',
53
+ 'path': 'path',
54
+ '_query_params': 'queryParams'
55
+ }
56
+
57
+ def __init__(self, body: 'V1Body' =None, method: 'str' =None, path: 'str' =None, _query_params: 'list[V1QueryParam]' =None): # noqa: E501
58
+ """V1DeploymentAPI - a model defined in Swagger""" # noqa: E501
59
+ self._body = None
60
+ self._method = None
61
+ self._path = None
62
+ self.__query_params = None
63
+ self.discriminator = None
64
+ if body is not None:
65
+ self.body = body
66
+ if method is not None:
67
+ self.method = method
68
+ if path is not None:
69
+ self.path = path
70
+ if _query_params is not None:
71
+ self._query_params = _query_params
72
+
73
+ @property
74
+ def body(self) -> 'V1Body':
75
+ """Gets the body of this V1DeploymentAPI. # noqa: E501
76
+
77
+
78
+ :return: The body of this V1DeploymentAPI. # noqa: E501
79
+ :rtype: V1Body
80
+ """
81
+ return self._body
82
+
83
+ @body.setter
84
+ def body(self, body: 'V1Body'):
85
+ """Sets the body of this V1DeploymentAPI.
86
+
87
+
88
+ :param body: The body of this V1DeploymentAPI. # noqa: E501
89
+ :type: V1Body
90
+ """
91
+
92
+ self._body = body
93
+
94
+ @property
95
+ def method(self) -> 'str':
96
+ """Gets the method of this V1DeploymentAPI. # noqa: E501
97
+
98
+
99
+ :return: The method of this V1DeploymentAPI. # noqa: E501
100
+ :rtype: str
101
+ """
102
+ return self._method
103
+
104
+ @method.setter
105
+ def method(self, method: 'str'):
106
+ """Sets the method of this V1DeploymentAPI.
107
+
108
+
109
+ :param method: The method of this V1DeploymentAPI. # noqa: E501
110
+ :type: str
111
+ """
112
+
113
+ self._method = method
114
+
115
+ @property
116
+ def path(self) -> 'str':
117
+ """Gets the path of this V1DeploymentAPI. # noqa: E501
118
+
119
+
120
+ :return: The path of this V1DeploymentAPI. # noqa: E501
121
+ :rtype: str
122
+ """
123
+ return self._path
124
+
125
+ @path.setter
126
+ def path(self, path: 'str'):
127
+ """Sets the path of this V1DeploymentAPI.
128
+
129
+
130
+ :param path: The path of this V1DeploymentAPI. # noqa: E501
131
+ :type: str
132
+ """
133
+
134
+ self._path = path
135
+
136
+ @property
137
+ def _query_params(self) -> 'list[V1QueryParam]':
138
+ """Gets the _query_params of this V1DeploymentAPI. # noqa: E501
139
+
140
+
141
+ :return: The _query_params of this V1DeploymentAPI. # noqa: E501
142
+ :rtype: list[V1QueryParam]
143
+ """
144
+ return self.__query_params
145
+
146
+ @_query_params.setter
147
+ def _query_params(self, _query_params: 'list[V1QueryParam]'):
148
+ """Sets the _query_params of this V1DeploymentAPI.
149
+
150
+
151
+ :param _query_params: The _query_params of this V1DeploymentAPI. # noqa: E501
152
+ :type: list[V1QueryParam]
153
+ """
154
+
155
+ self.__query_params = _query_params
156
+
157
+ def to_dict(self) -> dict:
158
+ """Returns the model properties as a dict"""
159
+ result = {}
160
+
161
+ for attr, _ in six.iteritems(self.swagger_types):
162
+ value = getattr(self, attr)
163
+ if isinstance(value, list):
164
+ result[attr] = list(map(
165
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
166
+ value
167
+ ))
168
+ elif hasattr(value, "to_dict"):
169
+ result[attr] = value.to_dict()
170
+ elif isinstance(value, dict):
171
+ result[attr] = dict(map(
172
+ lambda item: (item[0], item[1].to_dict())
173
+ if hasattr(item[1], "to_dict") else item,
174
+ value.items()
175
+ ))
176
+ else:
177
+ result[attr] = value
178
+ if issubclass(V1DeploymentAPI, dict):
179
+ for key, value in self.items():
180
+ result[key] = value
181
+
182
+ return result
183
+
184
+ def to_str(self) -> str:
185
+ """Returns the string representation of the model"""
186
+ return pprint.pformat(self.to_dict())
187
+
188
+ def __repr__(self) -> str:
189
+ """For `print` and `pprint`"""
190
+ return self.to_str()
191
+
192
+ def __eq__(self, other: 'V1DeploymentAPI') -> bool:
193
+ """Returns true if both objects are equal"""
194
+ if not isinstance(other, V1DeploymentAPI):
195
+ return False
196
+
197
+ return self.__dict__ == other.__dict__
198
+
199
+ def __ne__(self, other: 'V1DeploymentAPI') -> bool:
200
+ """Returns true if both objects are not equal"""
201
+ return not self == other
@@ -42,6 +42,7 @@ class V1GetModelFilesResponse(object):
42
42
  """
43
43
  swagger_types = {
44
44
  'filepaths': 'list[str]',
45
+ 'filesizes': 'list[str]',
45
46
  'model_id': 'str',
46
47
  'project_id': 'str',
47
48
  'size_bytes': 'str',
@@ -51,6 +52,7 @@ class V1GetModelFilesResponse(object):
51
52
 
52
53
  attribute_map = {
53
54
  'filepaths': 'filepaths',
55
+ 'filesizes': 'filesizes',
54
56
  'model_id': 'modelId',
55
57
  'project_id': 'projectId',
56
58
  'size_bytes': 'sizeBytes',
@@ -58,9 +60,10 @@ class V1GetModelFilesResponse(object):
58
60
  'version': 'version'
59
61
  }
60
62
 
61
- def __init__(self, filepaths: 'list[str]' =None, model_id: 'str' =None, project_id: 'str' =None, size_bytes: 'str' =None, upload_complete: 'bool' =None, version: 'str' =None): # noqa: E501
63
+ def __init__(self, filepaths: 'list[str]' =None, filesizes: 'list[str]' =None, model_id: 'str' =None, project_id: 'str' =None, size_bytes: 'str' =None, upload_complete: 'bool' =None, version: 'str' =None): # noqa: E501
62
64
  """V1GetModelFilesResponse - a model defined in Swagger""" # noqa: E501
63
65
  self._filepaths = None
66
+ self._filesizes = None
64
67
  self._model_id = None
65
68
  self._project_id = None
66
69
  self._size_bytes = None
@@ -69,6 +72,8 @@ class V1GetModelFilesResponse(object):
69
72
  self.discriminator = None
70
73
  if filepaths is not None:
71
74
  self.filepaths = filepaths
75
+ if filesizes is not None:
76
+ self.filesizes = filesizes
72
77
  if model_id is not None:
73
78
  self.model_id = model_id
74
79
  if project_id is not None:
@@ -101,6 +106,27 @@ class V1GetModelFilesResponse(object):
101
106
 
102
107
  self._filepaths = filepaths
103
108
 
109
+ @property
110
+ def filesizes(self) -> 'list[str]':
111
+ """Gets the filesizes of this V1GetModelFilesResponse. # noqa: E501
112
+
113
+
114
+ :return: The filesizes of this V1GetModelFilesResponse. # noqa: E501
115
+ :rtype: list[str]
116
+ """
117
+ return self._filesizes
118
+
119
+ @filesizes.setter
120
+ def filesizes(self, filesizes: 'list[str]'):
121
+ """Sets the filesizes of this V1GetModelFilesResponse.
122
+
123
+
124
+ :param filesizes: The filesizes of this V1GetModelFilesResponse. # noqa: E501
125
+ :type: list[str]
126
+ """
127
+
128
+ self._filesizes = filesizes
129
+
104
130
  @property
105
131
  def model_id(self) -> 'str':
106
132
  """Gets the model_id of this V1GetModelFilesResponse. # noqa: E501