lightning-sdk 0.1.41__py3-none-any.whl → 0.1.42__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 (74) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/ai_hub.py +8 -3
  3. lightning_sdk/api/ai_hub_api.py +3 -3
  4. lightning_sdk/api/deployment_api.py +6 -6
  5. lightning_sdk/api/job_api.py +32 -6
  6. lightning_sdk/api/mmt_api.py +59 -19
  7. lightning_sdk/api/studio_api.py +37 -19
  8. lightning_sdk/api/teamspace_api.py +34 -29
  9. lightning_sdk/api/utils.py +46 -34
  10. lightning_sdk/cli/ai_hub.py +3 -3
  11. lightning_sdk/cli/entrypoint.py +3 -1
  12. lightning_sdk/cli/mmt.py +11 -10
  13. lightning_sdk/cli/run.py +9 -8
  14. lightning_sdk/cli/serve.py +130 -0
  15. lightning_sdk/deployment/deployment.py +18 -12
  16. lightning_sdk/job/base.py +118 -24
  17. lightning_sdk/job/job.py +87 -9
  18. lightning_sdk/job/v1.py +75 -18
  19. lightning_sdk/job/v2.py +51 -15
  20. lightning_sdk/job/work.py +36 -7
  21. lightning_sdk/lightning_cloud/openapi/__init__.py +12 -0
  22. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +215 -5
  23. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +218 -0
  24. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +226 -0
  25. lightning_sdk/lightning_cloud/openapi/api/snowflake_service_api.py +21 -1
  26. lightning_sdk/lightning_cloud/openapi/models/__init__.py +12 -0
  27. lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/id_visibility_body.py +123 -0
  29. lightning_sdk/lightning_cloud/openapi/models/model_id_versions_body.py +29 -3
  30. lightning_sdk/lightning_cloud/openapi/models/project_id_multimachinejobs_body.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/project_id_snowflake_body.py +15 -67
  32. lightning_sdk/lightning_cloud/openapi/models/query_query_id_body.py +17 -69
  33. lightning_sdk/lightning_cloud/openapi/models/snowflake_export_body.py +29 -81
  34. lightning_sdk/lightning_cloud/openapi/models/snowflake_query_body.py +17 -69
  35. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_file_url_response.py +27 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py +17 -17
  37. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_url_response.py +149 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_get_project_balance_response.py +27 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_list_multi_machine_job_events_response.py +123 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +27 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_model_file.py +175 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +27 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event.py +331 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event_type.py +104 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance.py +149 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance_strategy.py +105 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py +27 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_snowflake_data_connection.py +29 -81
  50. lightning_sdk/lightning_cloud/openapi/models/v1_system_metrics.py +29 -3
  51. lightning_sdk/lightning_cloud/openapi/models/v1_trainium_system_metrics.py +175 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_update_metrics_stream_visibility_response.py +97 -0
  53. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +27 -53
  54. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_request.py +149 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_response.py +97 -0
  56. lightning_sdk/lightning_cloud/rest_client.py +2 -0
  57. lightning_sdk/mmt/__init__.py +3 -0
  58. lightning_sdk/{_mmt → mmt}/base.py +20 -14
  59. lightning_sdk/{_mmt → mmt}/mmt.py +46 -17
  60. lightning_sdk/mmt/v1.py +129 -0
  61. lightning_sdk/{_mmt → mmt}/v2.py +16 -21
  62. lightning_sdk/plugin.py +43 -16
  63. lightning_sdk/services/file_endpoint.py +11 -5
  64. lightning_sdk/studio.py +16 -9
  65. lightning_sdk/teamspace.py +21 -8
  66. lightning_sdk/utils/resolve.py +18 -0
  67. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.42.dist-info}/METADATA +3 -1
  68. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.42.dist-info}/RECORD +72 -59
  69. lightning_sdk/_mmt/__init__.py +0 -3
  70. lightning_sdk/_mmt/v1.py +0 -69
  71. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.42.dist-info}/LICENSE +0 -0
  72. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.42.dist-info}/WHEEL +0 -0
  73. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.42.dist-info}/entry_points.txt +0 -0
  74. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.42.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,175 @@
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 V1TrainiumSystemMetrics(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
+ 'memory_total': 'int',
45
+ 'memory_used': 'int',
46
+ 'utilization': 'int'
47
+ }
48
+
49
+ attribute_map = {
50
+ 'memory_total': 'memoryTotal',
51
+ 'memory_used': 'memoryUsed',
52
+ 'utilization': 'utilization'
53
+ }
54
+
55
+ def __init__(self, memory_total: 'int' =None, memory_used: 'int' =None, utilization: 'int' =None): # noqa: E501
56
+ """V1TrainiumSystemMetrics - a model defined in Swagger""" # noqa: E501
57
+ self._memory_total = None
58
+ self._memory_used = None
59
+ self._utilization = None
60
+ self.discriminator = None
61
+ if memory_total is not None:
62
+ self.memory_total = memory_total
63
+ if memory_used is not None:
64
+ self.memory_used = memory_used
65
+ if utilization is not None:
66
+ self.utilization = utilization
67
+
68
+ @property
69
+ def memory_total(self) -> 'int':
70
+ """Gets the memory_total of this V1TrainiumSystemMetrics. # noqa: E501
71
+
72
+
73
+ :return: The memory_total of this V1TrainiumSystemMetrics. # noqa: E501
74
+ :rtype: int
75
+ """
76
+ return self._memory_total
77
+
78
+ @memory_total.setter
79
+ def memory_total(self, memory_total: 'int'):
80
+ """Sets the memory_total of this V1TrainiumSystemMetrics.
81
+
82
+
83
+ :param memory_total: The memory_total of this V1TrainiumSystemMetrics. # noqa: E501
84
+ :type: int
85
+ """
86
+
87
+ self._memory_total = memory_total
88
+
89
+ @property
90
+ def memory_used(self) -> 'int':
91
+ """Gets the memory_used of this V1TrainiumSystemMetrics. # noqa: E501
92
+
93
+
94
+ :return: The memory_used of this V1TrainiumSystemMetrics. # noqa: E501
95
+ :rtype: int
96
+ """
97
+ return self._memory_used
98
+
99
+ @memory_used.setter
100
+ def memory_used(self, memory_used: 'int'):
101
+ """Sets the memory_used of this V1TrainiumSystemMetrics.
102
+
103
+
104
+ :param memory_used: The memory_used of this V1TrainiumSystemMetrics. # noqa: E501
105
+ :type: int
106
+ """
107
+
108
+ self._memory_used = memory_used
109
+
110
+ @property
111
+ def utilization(self) -> 'int':
112
+ """Gets the utilization of this V1TrainiumSystemMetrics. # noqa: E501
113
+
114
+
115
+ :return: The utilization of this V1TrainiumSystemMetrics. # noqa: E501
116
+ :rtype: int
117
+ """
118
+ return self._utilization
119
+
120
+ @utilization.setter
121
+ def utilization(self, utilization: 'int'):
122
+ """Sets the utilization of this V1TrainiumSystemMetrics.
123
+
124
+
125
+ :param utilization: The utilization of this V1TrainiumSystemMetrics. # noqa: E501
126
+ :type: int
127
+ """
128
+
129
+ self._utilization = utilization
130
+
131
+ def to_dict(self) -> dict:
132
+ """Returns the model properties as a dict"""
133
+ result = {}
134
+
135
+ for attr, _ in six.iteritems(self.swagger_types):
136
+ value = getattr(self, attr)
137
+ if isinstance(value, list):
138
+ result[attr] = list(map(
139
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
140
+ value
141
+ ))
142
+ elif hasattr(value, "to_dict"):
143
+ result[attr] = value.to_dict()
144
+ elif isinstance(value, dict):
145
+ result[attr] = dict(map(
146
+ lambda item: (item[0], item[1].to_dict())
147
+ if hasattr(item[1], "to_dict") else item,
148
+ value.items()
149
+ ))
150
+ else:
151
+ result[attr] = value
152
+ if issubclass(V1TrainiumSystemMetrics, dict):
153
+ for key, value in self.items():
154
+ result[key] = value
155
+
156
+ return result
157
+
158
+ def to_str(self) -> str:
159
+ """Returns the string representation of the model"""
160
+ return pprint.pformat(self.to_dict())
161
+
162
+ def __repr__(self) -> str:
163
+ """For `print` and `pprint`"""
164
+ return self.to_str()
165
+
166
+ def __eq__(self, other: 'V1TrainiumSystemMetrics') -> bool:
167
+ """Returns true if both objects are equal"""
168
+ if not isinstance(other, V1TrainiumSystemMetrics):
169
+ return False
170
+
171
+ return self.__dict__ == other.__dict__
172
+
173
+ def __ne__(self, other: 'V1TrainiumSystemMetrics') -> bool:
174
+ """Returns true if both objects are not equal"""
175
+ return not self == other
@@ -0,0 +1,97 @@
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 V1UpdateMetricsStreamVisibilityResponse(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
+ }
45
+
46
+ attribute_map = {
47
+ }
48
+
49
+ def __init__(self): # noqa: E501
50
+ """V1UpdateMetricsStreamVisibilityResponse - a model defined in Swagger""" # noqa: E501
51
+ self.discriminator = None
52
+
53
+ def to_dict(self) -> dict:
54
+ """Returns the model properties as a dict"""
55
+ result = {}
56
+
57
+ for attr, _ in six.iteritems(self.swagger_types):
58
+ value = getattr(self, attr)
59
+ if isinstance(value, list):
60
+ result[attr] = list(map(
61
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
62
+ value
63
+ ))
64
+ elif hasattr(value, "to_dict"):
65
+ result[attr] = value.to_dict()
66
+ elif isinstance(value, dict):
67
+ result[attr] = dict(map(
68
+ lambda item: (item[0], item[1].to_dict())
69
+ if hasattr(item[1], "to_dict") else item,
70
+ value.items()
71
+ ))
72
+ else:
73
+ result[attr] = value
74
+ if issubclass(V1UpdateMetricsStreamVisibilityResponse, dict):
75
+ for key, value in self.items():
76
+ result[key] = value
77
+
78
+ return result
79
+
80
+ def to_str(self) -> str:
81
+ """Returns the string representation of the model"""
82
+ return pprint.pformat(self.to_dict())
83
+
84
+ def __repr__(self) -> str:
85
+ """For `print` and `pprint`"""
86
+ return self.to_str()
87
+
88
+ def __eq__(self, other: 'V1UpdateMetricsStreamVisibilityResponse') -> bool:
89
+ """Returns true if both objects are equal"""
90
+ if not isinstance(other, V1UpdateMetricsStreamVisibilityResponse):
91
+ return False
92
+
93
+ return self.__dict__ == other.__dict__
94
+
95
+ def __ne__(self, other: 'V1UpdateMetricsStreamVisibilityResponse') -> bool:
96
+ """Returns true if both objects are not equal"""
97
+ return not self == other
@@ -47,7 +47,6 @@ class V1UserFeatures(object):
47
47
  'ai_hub_monetization': 'bool',
48
48
  'auto_fast_load': 'bool',
49
49
  'auto_join_orgs': 'bool',
50
- 'aws_trainium': 'bool',
51
50
  'b2c_experience': 'bool',
52
51
  'cap_add': 'list[str]',
53
52
  'cap_drop': 'list[str]',
@@ -57,7 +56,6 @@ class V1UserFeatures(object):
57
56
  'code_tab': 'bool',
58
57
  'collab_screen_sharing': 'bool',
59
58
  'cost_attribution_settings': 'bool',
60
- 'crypto_monitoring': 'bool',
61
59
  'custom_app_domain': 'bool',
62
60
  'custom_instance_types': 'bool',
63
61
  'default_one_cluster': 'bool',
@@ -85,6 +83,7 @@ class V1UserFeatures(object):
85
83
  'lit_logger_storage_v2': 'bool',
86
84
  'mmt_fault_tolerance': 'bool',
87
85
  'mmt_strategy_selector': 'bool',
86
+ 'mmt_v2': 'bool',
88
87
  'model_store': 'bool',
89
88
  'multiple_deployment_versions': 'bool',
90
89
  'multiple_studio_versions': 'bool',
@@ -132,7 +131,6 @@ class V1UserFeatures(object):
132
131
  'ai_hub_monetization': 'aiHubMonetization',
133
132
  'auto_fast_load': 'autoFastLoad',
134
133
  'auto_join_orgs': 'autoJoinOrgs',
135
- 'aws_trainium': 'awsTrainium',
136
134
  'b2c_experience': 'b2cExperience',
137
135
  'cap_add': 'capAdd',
138
136
  'cap_drop': 'capDrop',
@@ -142,7 +140,6 @@ class V1UserFeatures(object):
142
140
  'code_tab': 'codeTab',
143
141
  'collab_screen_sharing': 'collabScreenSharing',
144
142
  'cost_attribution_settings': 'costAttributionSettings',
145
- 'crypto_monitoring': 'cryptoMonitoring',
146
143
  'custom_app_domain': 'customAppDomain',
147
144
  'custom_instance_types': 'customInstanceTypes',
148
145
  'default_one_cluster': 'defaultOneCluster',
@@ -170,6 +167,7 @@ class V1UserFeatures(object):
170
167
  'lit_logger_storage_v2': 'litLoggerStorageV2',
171
168
  'mmt_fault_tolerance': 'mmtFaultTolerance',
172
169
  'mmt_strategy_selector': 'mmtStrategySelector',
170
+ 'mmt_v2': 'mmtV2',
173
171
  'model_store': 'modelStore',
174
172
  'multiple_deployment_versions': 'multipleDeploymentVersions',
175
173
  'multiple_studio_versions': 'multipleStudioVersions',
@@ -210,7 +208,7 @@ class V1UserFeatures(object):
210
208
  'writable_data_connections': 'writableDataConnections'
211
209
  }
212
210
 
213
- def __init__(self, advanced_deployment_autoscaling: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, aws_trainium: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, cluster_proxy: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, crypto_monitoring: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, default_one_cluster: 'bool' =None, deployment_customize_api: 'bool' =None, deployment_data_path: 'bool' =None, deployment_gallery: 'bool' =None, deployment_persistent_disk: 'bool' =None, deployment_version_visibility: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, dynamic_workload_scheduling: 'bool' =None, enable_crypto_crackdown: 'bool' =None, enable_efs: 'bool' =None, enable_storage_limits: 'bool' =None, featured_studios_admin: 'bool' =None, filesystem_optimisation: 'bool' =None, gcp: 'bool' =None, inference_job_deployment_plugin: 'bool' =None, instant_capacity_reservation: 'bool' =None, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: 'bool' =None, lightning_registry: 'bool' =None, lit_logger: 'bool' =None, lit_logger_storage_v2: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_store: 'bool' =None, multiple_deployment_versions: 'bool' =None, multiple_studio_versions: 'bool' =None, org_level_member_permissions: 'bool' =None, plugin_biz_chat: 'bool' =None, plugin_distributed: 'bool' =None, plugin_fiftyone: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_lightning_apps: 'bool' =None, plugin_lightning_apps_distributed: 'bool' =None, plugin_mage_ai: 'bool' =None, plugin_milvus: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_react: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, plugin_weviate: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, restart_ide_on_hang: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, s3_folders: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snapshotter_service: 'bool' =None, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, use_rclone_mounts_only: 'bool' =None, writable_data_connections: 'bool' =None): # noqa: E501
211
+ def __init__(self, advanced_deployment_autoscaling: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, cluster_proxy: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, default_one_cluster: 'bool' =None, deployment_customize_api: 'bool' =None, deployment_data_path: 'bool' =None, deployment_gallery: 'bool' =None, deployment_persistent_disk: 'bool' =None, deployment_version_visibility: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, dynamic_workload_scheduling: 'bool' =None, enable_crypto_crackdown: 'bool' =None, enable_efs: 'bool' =None, enable_storage_limits: 'bool' =None, featured_studios_admin: 'bool' =None, filesystem_optimisation: 'bool' =None, gcp: 'bool' =None, inference_job_deployment_plugin: 'bool' =None, instant_capacity_reservation: 'bool' =None, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: 'bool' =None, lightning_registry: 'bool' =None, lit_logger: 'bool' =None, lit_logger_storage_v2: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, mmt_v2: 'bool' =None, model_store: 'bool' =None, multiple_deployment_versions: 'bool' =None, multiple_studio_versions: 'bool' =None, org_level_member_permissions: 'bool' =None, plugin_biz_chat: 'bool' =None, plugin_distributed: 'bool' =None, plugin_fiftyone: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_lightning_apps: 'bool' =None, plugin_lightning_apps_distributed: 'bool' =None, plugin_mage_ai: 'bool' =None, plugin_milvus: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_react: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, plugin_weviate: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, restart_ide_on_hang: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, s3_folders: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snapshotter_service: 'bool' =None, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, use_rclone_mounts_only: 'bool' =None, writable_data_connections: 'bool' =None): # noqa: E501
214
212
  """V1UserFeatures - a model defined in Swagger""" # noqa: E501
215
213
  self._advanced_deployment_autoscaling = None
216
214
  self._affiliate_links = None
@@ -218,7 +216,6 @@ class V1UserFeatures(object):
218
216
  self._ai_hub_monetization = None
219
217
  self._auto_fast_load = None
220
218
  self._auto_join_orgs = None
221
- self._aws_trainium = None
222
219
  self._b2c_experience = None
223
220
  self._cap_add = None
224
221
  self._cap_drop = None
@@ -228,7 +225,6 @@ class V1UserFeatures(object):
228
225
  self._code_tab = None
229
226
  self._collab_screen_sharing = None
230
227
  self._cost_attribution_settings = None
231
- self._crypto_monitoring = None
232
228
  self._custom_app_domain = None
233
229
  self._custom_instance_types = None
234
230
  self._default_one_cluster = None
@@ -256,6 +252,7 @@ class V1UserFeatures(object):
256
252
  self._lit_logger_storage_v2 = None
257
253
  self._mmt_fault_tolerance = None
258
254
  self._mmt_strategy_selector = None
255
+ self._mmt_v2 = None
259
256
  self._model_store = None
260
257
  self._multiple_deployment_versions = None
261
258
  self._multiple_studio_versions = None
@@ -307,8 +304,6 @@ class V1UserFeatures(object):
307
304
  self.auto_fast_load = auto_fast_load
308
305
  if auto_join_orgs is not None:
309
306
  self.auto_join_orgs = auto_join_orgs
310
- if aws_trainium is not None:
311
- self.aws_trainium = aws_trainium
312
307
  if b2c_experience is not None:
313
308
  self.b2c_experience = b2c_experience
314
309
  if cap_add is not None:
@@ -327,8 +322,6 @@ class V1UserFeatures(object):
327
322
  self.collab_screen_sharing = collab_screen_sharing
328
323
  if cost_attribution_settings is not None:
329
324
  self.cost_attribution_settings = cost_attribution_settings
330
- if crypto_monitoring is not None:
331
- self.crypto_monitoring = crypto_monitoring
332
325
  if custom_app_domain is not None:
333
326
  self.custom_app_domain = custom_app_domain
334
327
  if custom_instance_types is not None:
@@ -383,6 +376,8 @@ class V1UserFeatures(object):
383
376
  self.mmt_fault_tolerance = mmt_fault_tolerance
384
377
  if mmt_strategy_selector is not None:
385
378
  self.mmt_strategy_selector = mmt_strategy_selector
379
+ if mmt_v2 is not None:
380
+ self.mmt_v2 = mmt_v2
386
381
  if model_store is not None:
387
382
  self.model_store = model_store
388
383
  if multiple_deployment_versions is not None:
@@ -586,27 +581,6 @@ class V1UserFeatures(object):
586
581
 
587
582
  self._auto_join_orgs = auto_join_orgs
588
583
 
589
- @property
590
- def aws_trainium(self) -> 'bool':
591
- """Gets the aws_trainium of this V1UserFeatures. # noqa: E501
592
-
593
-
594
- :return: The aws_trainium of this V1UserFeatures. # noqa: E501
595
- :rtype: bool
596
- """
597
- return self._aws_trainium
598
-
599
- @aws_trainium.setter
600
- def aws_trainium(self, aws_trainium: 'bool'):
601
- """Sets the aws_trainium of this V1UserFeatures.
602
-
603
-
604
- :param aws_trainium: The aws_trainium of this V1UserFeatures. # noqa: E501
605
- :type: bool
606
- """
607
-
608
- self._aws_trainium = aws_trainium
609
-
610
584
  @property
611
585
  def b2c_experience(self) -> 'bool':
612
586
  """Gets the b2c_experience of this V1UserFeatures. # noqa: E501
@@ -796,27 +770,6 @@ class V1UserFeatures(object):
796
770
 
797
771
  self._cost_attribution_settings = cost_attribution_settings
798
772
 
799
- @property
800
- def crypto_monitoring(self) -> 'bool':
801
- """Gets the crypto_monitoring of this V1UserFeatures. # noqa: E501
802
-
803
-
804
- :return: The crypto_monitoring of this V1UserFeatures. # noqa: E501
805
- :rtype: bool
806
- """
807
- return self._crypto_monitoring
808
-
809
- @crypto_monitoring.setter
810
- def crypto_monitoring(self, crypto_monitoring: 'bool'):
811
- """Sets the crypto_monitoring of this V1UserFeatures.
812
-
813
-
814
- :param crypto_monitoring: The crypto_monitoring of this V1UserFeatures. # noqa: E501
815
- :type: bool
816
- """
817
-
818
- self._crypto_monitoring = crypto_monitoring
819
-
820
773
  @property
821
774
  def custom_app_domain(self) -> 'bool':
822
775
  """Gets the custom_app_domain of this V1UserFeatures. # noqa: E501
@@ -1384,6 +1337,27 @@ class V1UserFeatures(object):
1384
1337
 
1385
1338
  self._mmt_strategy_selector = mmt_strategy_selector
1386
1339
 
1340
+ @property
1341
+ def mmt_v2(self) -> 'bool':
1342
+ """Gets the mmt_v2 of this V1UserFeatures. # noqa: E501
1343
+
1344
+
1345
+ :return: The mmt_v2 of this V1UserFeatures. # noqa: E501
1346
+ :rtype: bool
1347
+ """
1348
+ return self._mmt_v2
1349
+
1350
+ @mmt_v2.setter
1351
+ def mmt_v2(self, mmt_v2: 'bool'):
1352
+ """Sets the mmt_v2 of this V1UserFeatures.
1353
+
1354
+
1355
+ :param mmt_v2: The mmt_v2 of this V1UserFeatures. # noqa: E501
1356
+ :type: bool
1357
+ """
1358
+
1359
+ self._mmt_v2 = mmt_v2
1360
+
1387
1361
  @property
1388
1362
  def model_store(self) -> 'bool':
1389
1363
  """Gets the model_store of this V1UserFeatures. # noqa: E501
@@ -0,0 +1,149 @@
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 V1ValidateDeploymentImageRequest(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
+ 'image': 'str',
45
+ 'secret': 'str'
46
+ }
47
+
48
+ attribute_map = {
49
+ 'image': 'image',
50
+ 'secret': 'secret'
51
+ }
52
+
53
+ def __init__(self, image: 'str' =None, secret: 'str' =None): # noqa: E501
54
+ """V1ValidateDeploymentImageRequest - a model defined in Swagger""" # noqa: E501
55
+ self._image = None
56
+ self._secret = None
57
+ self.discriminator = None
58
+ if image is not None:
59
+ self.image = image
60
+ if secret is not None:
61
+ self.secret = secret
62
+
63
+ @property
64
+ def image(self) -> 'str':
65
+ """Gets the image of this V1ValidateDeploymentImageRequest. # noqa: E501
66
+
67
+
68
+ :return: The image of this V1ValidateDeploymentImageRequest. # noqa: E501
69
+ :rtype: str
70
+ """
71
+ return self._image
72
+
73
+ @image.setter
74
+ def image(self, image: 'str'):
75
+ """Sets the image of this V1ValidateDeploymentImageRequest.
76
+
77
+
78
+ :param image: The image of this V1ValidateDeploymentImageRequest. # noqa: E501
79
+ :type: str
80
+ """
81
+
82
+ self._image = image
83
+
84
+ @property
85
+ def secret(self) -> 'str':
86
+ """Gets the secret of this V1ValidateDeploymentImageRequest. # noqa: E501
87
+
88
+
89
+ :return: The secret of this V1ValidateDeploymentImageRequest. # noqa: E501
90
+ :rtype: str
91
+ """
92
+ return self._secret
93
+
94
+ @secret.setter
95
+ def secret(self, secret: 'str'):
96
+ """Sets the secret of this V1ValidateDeploymentImageRequest.
97
+
98
+
99
+ :param secret: The secret of this V1ValidateDeploymentImageRequest. # noqa: E501
100
+ :type: str
101
+ """
102
+
103
+ self._secret = secret
104
+
105
+ def to_dict(self) -> dict:
106
+ """Returns the model properties as a dict"""
107
+ result = {}
108
+
109
+ for attr, _ in six.iteritems(self.swagger_types):
110
+ value = getattr(self, attr)
111
+ if isinstance(value, list):
112
+ result[attr] = list(map(
113
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
114
+ value
115
+ ))
116
+ elif hasattr(value, "to_dict"):
117
+ result[attr] = value.to_dict()
118
+ elif isinstance(value, dict):
119
+ result[attr] = dict(map(
120
+ lambda item: (item[0], item[1].to_dict())
121
+ if hasattr(item[1], "to_dict") else item,
122
+ value.items()
123
+ ))
124
+ else:
125
+ result[attr] = value
126
+ if issubclass(V1ValidateDeploymentImageRequest, dict):
127
+ for key, value in self.items():
128
+ result[key] = value
129
+
130
+ return result
131
+
132
+ def to_str(self) -> str:
133
+ """Returns the string representation of the model"""
134
+ return pprint.pformat(self.to_dict())
135
+
136
+ def __repr__(self) -> str:
137
+ """For `print` and `pprint`"""
138
+ return self.to_str()
139
+
140
+ def __eq__(self, other: 'V1ValidateDeploymentImageRequest') -> bool:
141
+ """Returns true if both objects are equal"""
142
+ if not isinstance(other, V1ValidateDeploymentImageRequest):
143
+ return False
144
+
145
+ return self.__dict__ == other.__dict__
146
+
147
+ def __ne__(self, other: 'V1ValidateDeploymentImageRequest') -> bool:
148
+ """Returns true if both objects are not equal"""
149
+ return not self == other