lightning-sdk 0.2.9__py3-none-any.whl → 0.2.10__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 (50) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/deployment_api.py +1 -0
  3. lightning_sdk/api/lit_container_api.py +19 -2
  4. lightning_sdk/api/teamspace_api.py +20 -16
  5. lightning_sdk/api/utils.py +1 -1
  6. lightning_sdk/cli/entrypoint.py +2 -2
  7. lightning_sdk/cli/serve.py +141 -16
  8. lightning_sdk/cli/upload.py +4 -1
  9. lightning_sdk/deployment/deployment.py +5 -2
  10. lightning_sdk/lightning_cloud/openapi/__init__.py +5 -0
  11. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +98 -1
  12. lightning_sdk/lightning_cloud/openapi/models/__init__.py +5 -0
  13. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityreservations_body.py +55 -3
  14. lightning_sdk/lightning_cloud/openapi/models/create.py +53 -1
  15. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +55 -3
  16. lightning_sdk/lightning_cloud/openapi/models/update.py +27 -1
  17. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +53 -1
  18. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_cold_start_metrics_stats.py +357 -0
  19. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  20. lightning_sdk/lightning_cloud/openapi/models/v1_cloudflare_v1.py +227 -0
  21. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +27 -1
  22. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +55 -3
  23. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +53 -1
  24. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  25. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  26. lightning_sdk/lightning_cloud/openapi/models/v1_create_cluster_capacity_reservation_response.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/v1_gcp_direct_vpc.py +149 -0
  29. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_cold_start_metrics_stats_response.py +123 -0
  30. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +53 -1
  31. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +17 -17
  32. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +55 -3
  33. lightning_sdk/lightning_cloud/openapi/models/v1_project_cluster_binding.py +27 -1
  34. lightning_sdk/lightning_cloud/openapi/models/v1_r2_data_connection.py +253 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +50 -24
  37. lightning_sdk/lightning_cloud/openapi/models/v1_validate_data_connection_response.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_weka_data_connection.py +29 -55
  39. lightning_sdk/lightning_cloud/openapi/models/validate.py +27 -1
  40. lightning_sdk/lit_container.py +12 -2
  41. lightning_sdk/models.py +22 -9
  42. lightning_sdk/serve.py +3 -0
  43. lightning_sdk/teamspace.py +2 -0
  44. lightning_sdk/utils/resolve.py +11 -4
  45. {lightning_sdk-0.2.9.dist-info → lightning_sdk-0.2.10.dist-info}/METADATA +1 -1
  46. {lightning_sdk-0.2.9.dist-info → lightning_sdk-0.2.10.dist-info}/RECORD +50 -45
  47. {lightning_sdk-0.2.9.dist-info → lightning_sdk-0.2.10.dist-info}/LICENSE +0 -0
  48. {lightning_sdk-0.2.9.dist-info → lightning_sdk-0.2.10.dist-info}/WHEEL +0 -0
  49. {lightning_sdk-0.2.9.dist-info → lightning_sdk-0.2.10.dist-info}/entry_points.txt +0 -0
  50. {lightning_sdk-0.2.9.dist-info → lightning_sdk-0.2.10.dist-info}/top_level.txt +0 -0
@@ -236,6 +236,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_artifact_event_
236
236
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version import V1CloudSpaceCodeVersion
237
237
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version_status import V1CloudSpaceCodeVersionStatus
238
238
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_cold_start_metrics import V1CloudSpaceColdStartMetrics
239
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_cold_start_metrics_stats import V1CloudSpaceColdStartMetricsStats
239
240
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_engagement_response import V1CloudSpaceEngagementResponse
240
241
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template import V1CloudSpaceEnvironmentTemplate
241
242
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template_config import V1CloudSpaceEnvironmentTemplateConfig
@@ -251,6 +252,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import
251
252
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
252
253
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
253
254
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
255
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloudflare_v1 import V1CloudflareV1
254
256
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_accelerator import V1ClusterAccelerator
255
257
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_availability import V1ClusterAvailability
256
258
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_capacity_reservation import V1ClusterCapacityReservation
@@ -442,6 +444,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1Fi
442
444
  from lightning_sdk.lightning_cloud.openapi.models.v1_find_capacity_block_offering_response import V1FindCapacityBlockOfferingResponse
443
445
  from lightning_sdk.lightning_cloud.openapi.models.v1_flowserver import V1Flowserver
444
446
  from lightning_sdk.lightning_cloud.openapi.models.v1_folder_index_status import V1FolderIndexStatus
447
+ from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_direct_vpc import V1GCPDirectVPC
445
448
  from lightning_sdk.lightning_cloud.openapi.models.v1_gcs_folder_data_connection import V1GCSFolderDataConnection
446
449
  from lightning_sdk.lightning_cloud.openapi.models.v1_gpu_system_metrics import V1GPUSystemMetrics
447
450
  from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_app import V1GalleryApp
@@ -453,6 +456,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_affiliate_link_response
453
456
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_env_response import V1GetAgentJobEnvResponse
454
457
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_logs_metadata_response import V1GetAgentJobLogsMetadataResponse
455
458
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_artifacts_page_response import V1GetArtifactsPageResponse
459
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_cold_start_metrics_stats_response import V1GetCloudSpaceColdStartMetricsStatsResponse
456
460
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_status_response import V1GetCloudSpaceInstanceStatusResponse
457
461
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_system_metrics_aggregate_response import V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
458
462
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
@@ -728,6 +732,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_quest_status import V1Quest
728
732
  from lightning_sdk.lightning_cloud.openapi.models.v1_queue_server_type import V1QueueServerType
729
733
  from lightning_sdk.lightning_cloud.openapi.models.v1_quotas import V1Quotas
730
734
  from lightning_sdk.lightning_cloud.openapi.models.v1_quote_subscription_response import V1QuoteSubscriptionResponse
735
+ from lightning_sdk.lightning_cloud.openapi.models.v1_r2_data_connection import V1R2DataConnection
731
736
  from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_index_response import V1RefreshIndexResponse
732
737
  from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_path_response import V1RefreshPathResponse
733
738
  from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_request import V1RefreshRequest
@@ -44,31 +44,37 @@ class ClusterIdCapacityreservationsBody(object):
44
44
  'capacity_reservation_type': 'str',
45
45
  'cloud_provider_capacity_reservation_id': 'str',
46
46
  'end_time': 'datetime',
47
+ 'full_cloud_provider_reservation_string': 'str',
47
48
  'instance_type': 'str',
48
49
  'match_pattern': 'str',
49
50
  'region': 'str',
50
- 'start_time': 'datetime'
51
+ 'start_time': 'datetime',
52
+ 'zone': 'str'
51
53
  }
52
54
 
53
55
  attribute_map = {
54
56
  'capacity_reservation_type': 'capacityReservationType',
55
57
  'cloud_provider_capacity_reservation_id': 'cloudProviderCapacityReservationId',
56
58
  'end_time': 'endTime',
59
+ 'full_cloud_provider_reservation_string': 'fullCloudProviderReservationString',
57
60
  'instance_type': 'instanceType',
58
61
  'match_pattern': 'matchPattern',
59
62
  'region': 'region',
60
- 'start_time': 'startTime'
63
+ 'start_time': 'startTime',
64
+ 'zone': 'zone'
61
65
  }
62
66
 
63
- def __init__(self, capacity_reservation_type: 'str' =None, cloud_provider_capacity_reservation_id: 'str' =None, end_time: 'datetime' =None, instance_type: 'str' =None, match_pattern: 'str' =None, region: 'str' =None, start_time: 'datetime' =None): # noqa: E501
67
+ def __init__(self, capacity_reservation_type: 'str' =None, cloud_provider_capacity_reservation_id: 'str' =None, end_time: 'datetime' =None, full_cloud_provider_reservation_string: 'str' =None, instance_type: 'str' =None, match_pattern: 'str' =None, region: 'str' =None, start_time: 'datetime' =None, zone: 'str' =None): # noqa: E501
64
68
  """ClusterIdCapacityreservationsBody - a model defined in Swagger""" # noqa: E501
65
69
  self._capacity_reservation_type = None
66
70
  self._cloud_provider_capacity_reservation_id = None
67
71
  self._end_time = None
72
+ self._full_cloud_provider_reservation_string = None
68
73
  self._instance_type = None
69
74
  self._match_pattern = None
70
75
  self._region = None
71
76
  self._start_time = None
77
+ self._zone = None
72
78
  self.discriminator = None
73
79
  if capacity_reservation_type is not None:
74
80
  self.capacity_reservation_type = capacity_reservation_type
@@ -76,6 +82,8 @@ class ClusterIdCapacityreservationsBody(object):
76
82
  self.cloud_provider_capacity_reservation_id = cloud_provider_capacity_reservation_id
77
83
  if end_time is not None:
78
84
  self.end_time = end_time
85
+ if full_cloud_provider_reservation_string is not None:
86
+ self.full_cloud_provider_reservation_string = full_cloud_provider_reservation_string
79
87
  if instance_type is not None:
80
88
  self.instance_type = instance_type
81
89
  if match_pattern is not None:
@@ -84,6 +92,8 @@ class ClusterIdCapacityreservationsBody(object):
84
92
  self.region = region
85
93
  if start_time is not None:
86
94
  self.start_time = start_time
95
+ if zone is not None:
96
+ self.zone = zone
87
97
 
88
98
  @property
89
99
  def capacity_reservation_type(self) -> 'str':
@@ -148,6 +158,27 @@ class ClusterIdCapacityreservationsBody(object):
148
158
 
149
159
  self._end_time = end_time
150
160
 
161
+ @property
162
+ def full_cloud_provider_reservation_string(self) -> 'str':
163
+ """Gets the full_cloud_provider_reservation_string of this ClusterIdCapacityreservationsBody. # noqa: E501
164
+
165
+
166
+ :return: The full_cloud_provider_reservation_string of this ClusterIdCapacityreservationsBody. # noqa: E501
167
+ :rtype: str
168
+ """
169
+ return self._full_cloud_provider_reservation_string
170
+
171
+ @full_cloud_provider_reservation_string.setter
172
+ def full_cloud_provider_reservation_string(self, full_cloud_provider_reservation_string: 'str'):
173
+ """Sets the full_cloud_provider_reservation_string of this ClusterIdCapacityreservationsBody.
174
+
175
+
176
+ :param full_cloud_provider_reservation_string: The full_cloud_provider_reservation_string of this ClusterIdCapacityreservationsBody. # noqa: E501
177
+ :type: str
178
+ """
179
+
180
+ self._full_cloud_provider_reservation_string = full_cloud_provider_reservation_string
181
+
151
182
  @property
152
183
  def instance_type(self) -> 'str':
153
184
  """Gets the instance_type of this ClusterIdCapacityreservationsBody. # noqa: E501
@@ -232,6 +263,27 @@ class ClusterIdCapacityreservationsBody(object):
232
263
 
233
264
  self._start_time = start_time
234
265
 
266
+ @property
267
+ def zone(self) -> 'str':
268
+ """Gets the zone of this ClusterIdCapacityreservationsBody. # noqa: E501
269
+
270
+
271
+ :return: The zone of this ClusterIdCapacityreservationsBody. # noqa: E501
272
+ :rtype: str
273
+ """
274
+ return self._zone
275
+
276
+ @zone.setter
277
+ def zone(self, zone: 'str'):
278
+ """Sets the zone of this ClusterIdCapacityreservationsBody.
279
+
280
+
281
+ :param zone: The zone of this ClusterIdCapacityreservationsBody. # noqa: E501
282
+ :type: str
283
+ """
284
+
285
+ self._zone = zone
286
+
235
287
  def to_dict(self) -> dict:
236
288
  """Returns the model properties as a dict"""
237
289
  result = {}
@@ -53,9 +53,11 @@ class Create(object):
53
53
  'gcs_folder': 'V1GCSFolderDataConnection',
54
54
  'id': 'str',
55
55
  'name': 'str',
56
+ 'r2': 'V1R2DataConnection',
56
57
  'run_cmds': 'list[str]',
57
58
  's3_folder': 'V1S3FolderDataConnection',
58
59
  'snowflake': 'V1SnowflakeDataConnection',
60
+ 'weka': 'V1WekaDataConnection',
59
61
  'writable': 'bool'
60
62
  }
61
63
 
@@ -72,13 +74,15 @@ class Create(object):
72
74
  'gcs_folder': 'gcsFolder',
73
75
  'id': 'id',
74
76
  'name': 'name',
77
+ 'r2': 'r2',
75
78
  'run_cmds': 'runCmds',
76
79
  's3_folder': 's3Folder',
77
80
  'snowflake': 'snowflake',
81
+ 'weka': 'weka',
78
82
  'writable': 'writable'
79
83
  }
80
84
 
81
- def __init__(self, access_cluster_ids: 'list[str]' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, create_index: 'bool' =None, create_resources: 'bool' =None, efs: 'V1EfsConfig' =None, filestore: 'V1FilestoreDataConnection' =None, force: 'bool' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, name: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None): # noqa: E501
85
+ def __init__(self, access_cluster_ids: 'list[str]' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, create_index: 'bool' =None, create_resources: 'bool' =None, efs: 'V1EfsConfig' =None, filestore: 'V1FilestoreDataConnection' =None, force: 'bool' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, name: 'str' =None, r2: 'V1R2DataConnection' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, weka: 'V1WekaDataConnection' =None, writable: 'bool' =None): # noqa: E501
82
86
  """Create - a model defined in Swagger""" # noqa: E501
83
87
  self._access_cluster_ids = None
84
88
  self._aws = None
@@ -92,9 +96,11 @@ class Create(object):
92
96
  self._gcs_folder = None
93
97
  self._id = None
94
98
  self._name = None
99
+ self._r2 = None
95
100
  self._run_cmds = None
96
101
  self._s3_folder = None
97
102
  self._snowflake = None
103
+ self._weka = None
98
104
  self._writable = None
99
105
  self.discriminator = None
100
106
  if access_cluster_ids is not None:
@@ -121,12 +127,16 @@ class Create(object):
121
127
  self.id = id
122
128
  if name is not None:
123
129
  self.name = name
130
+ if r2 is not None:
131
+ self.r2 = r2
124
132
  if run_cmds is not None:
125
133
  self.run_cmds = run_cmds
126
134
  if s3_folder is not None:
127
135
  self.s3_folder = s3_folder
128
136
  if snowflake is not None:
129
137
  self.snowflake = snowflake
138
+ if weka is not None:
139
+ self.weka = weka
130
140
  if writable is not None:
131
141
  self.writable = writable
132
142
 
@@ -382,6 +392,27 @@ class Create(object):
382
392
 
383
393
  self._name = name
384
394
 
395
+ @property
396
+ def r2(self) -> 'V1R2DataConnection':
397
+ """Gets the r2 of this Create. # noqa: E501
398
+
399
+
400
+ :return: The r2 of this Create. # noqa: E501
401
+ :rtype: V1R2DataConnection
402
+ """
403
+ return self._r2
404
+
405
+ @r2.setter
406
+ def r2(self, r2: 'V1R2DataConnection'):
407
+ """Sets the r2 of this Create.
408
+
409
+
410
+ :param r2: The r2 of this Create. # noqa: E501
411
+ :type: V1R2DataConnection
412
+ """
413
+
414
+ self._r2 = r2
415
+
385
416
  @property
386
417
  def run_cmds(self) -> 'list[str]':
387
418
  """Gets the run_cmds of this Create. # noqa: E501
@@ -445,6 +476,27 @@ class Create(object):
445
476
 
446
477
  self._snowflake = snowflake
447
478
 
479
+ @property
480
+ def weka(self) -> 'V1WekaDataConnection':
481
+ """Gets the weka of this Create. # noqa: E501
482
+
483
+
484
+ :return: The weka of this Create. # noqa: E501
485
+ :rtype: V1WekaDataConnection
486
+ """
487
+ return self._weka
488
+
489
+ @weka.setter
490
+ def weka(self, weka: 'V1WekaDataConnection'):
491
+ """Sets the weka of this Create.
492
+
493
+
494
+ :param weka: The weka of this Create. # noqa: E501
495
+ :type: V1WekaDataConnection
496
+ """
497
+
498
+ self._weka = weka
499
+
448
500
  @property
449
501
  def writable(self) -> 'bool':
450
502
  """Gets the writable of this Create. # noqa: E501
@@ -43,6 +43,7 @@ class OrgsIdBody(object):
43
43
  swagger_types = {
44
44
  'alerts_config': 'V1AlertsConfig',
45
45
  'allow_aws_saas': 'bool',
46
+ 'allow_budgeting': 'bool',
46
47
  'allow_gcp_saas': 'bool',
47
48
  'allow_guest': 'bool',
48
49
  'allow_lambda_saas': 'bool',
@@ -64,12 +65,14 @@ class OrgsIdBody(object):
64
65
  'preferred_studio_provider': 'str',
65
66
  'start_studios_on_spot_instance': 'bool',
66
67
  'teamspace_default_credits': 'float',
67
- 'twitter_username': 'str'
68
+ 'twitter_username': 'str',
69
+ 'workload_max_run_duration': 'str'
68
70
  }
69
71
 
70
72
  attribute_map = {
71
73
  'alerts_config': 'alertsConfig',
72
74
  'allow_aws_saas': 'allowAwsSaas',
75
+ 'allow_budgeting': 'allowBudgeting',
73
76
  'allow_gcp_saas': 'allowGcpSaas',
74
77
  'allow_guest': 'allowGuest',
75
78
  'allow_lambda_saas': 'allowLambdaSaas',
@@ -91,13 +94,15 @@ class OrgsIdBody(object):
91
94
  'preferred_studio_provider': 'preferredStudioProvider',
92
95
  'start_studios_on_spot_instance': 'startStudiosOnSpotInstance',
93
96
  'teamspace_default_credits': 'teamspaceDefaultCredits',
94
- 'twitter_username': 'twitterUsername'
97
+ 'twitter_username': 'twitterUsername',
98
+ 'workload_max_run_duration': 'workloadMaxRunDuration'
95
99
  }
96
100
 
97
- def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_aws_saas: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domains: 'list[str]' =None, default_machine_type: 'str' =None, description: 'str' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, location: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None): # noqa: E501
101
+ def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_aws_saas: 'bool' =None, allow_budgeting: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domains: 'list[str]' =None, default_machine_type: 'str' =None, description: 'str' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, location: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, workload_max_run_duration: 'str' =None): # noqa: E501
98
102
  """OrgsIdBody - a model defined in Swagger""" # noqa: E501
99
103
  self._alerts_config = None
100
104
  self._allow_aws_saas = None
105
+ self._allow_budgeting = None
101
106
  self._allow_gcp_saas = None
102
107
  self._allow_guest = None
103
108
  self._allow_lambda_saas = None
@@ -120,11 +125,14 @@ class OrgsIdBody(object):
120
125
  self._start_studios_on_spot_instance = None
121
126
  self._teamspace_default_credits = None
122
127
  self._twitter_username = None
128
+ self._workload_max_run_duration = None
123
129
  self.discriminator = None
124
130
  if alerts_config is not None:
125
131
  self.alerts_config = alerts_config
126
132
  if allow_aws_saas is not None:
127
133
  self.allow_aws_saas = allow_aws_saas
134
+ if allow_budgeting is not None:
135
+ self.allow_budgeting = allow_budgeting
128
136
  if allow_gcp_saas is not None:
129
137
  self.allow_gcp_saas = allow_gcp_saas
130
138
  if allow_guest is not None:
@@ -169,6 +177,8 @@ class OrgsIdBody(object):
169
177
  self.teamspace_default_credits = teamspace_default_credits
170
178
  if twitter_username is not None:
171
179
  self.twitter_username = twitter_username
180
+ if workload_max_run_duration is not None:
181
+ self.workload_max_run_duration = workload_max_run_duration
172
182
 
173
183
  @property
174
184
  def alerts_config(self) -> 'V1AlertsConfig':
@@ -212,6 +222,27 @@ class OrgsIdBody(object):
212
222
 
213
223
  self._allow_aws_saas = allow_aws_saas
214
224
 
225
+ @property
226
+ def allow_budgeting(self) -> 'bool':
227
+ """Gets the allow_budgeting of this OrgsIdBody. # noqa: E501
228
+
229
+
230
+ :return: The allow_budgeting of this OrgsIdBody. # noqa: E501
231
+ :rtype: bool
232
+ """
233
+ return self._allow_budgeting
234
+
235
+ @allow_budgeting.setter
236
+ def allow_budgeting(self, allow_budgeting: 'bool'):
237
+ """Sets the allow_budgeting of this OrgsIdBody.
238
+
239
+
240
+ :param allow_budgeting: The allow_budgeting of this OrgsIdBody. # noqa: E501
241
+ :type: bool
242
+ """
243
+
244
+ self._allow_budgeting = allow_budgeting
245
+
215
246
  @property
216
247
  def allow_gcp_saas(self) -> 'bool':
217
248
  """Gets the allow_gcp_saas of this OrgsIdBody. # noqa: E501
@@ -674,6 +705,27 @@ class OrgsIdBody(object):
674
705
 
675
706
  self._twitter_username = twitter_username
676
707
 
708
+ @property
709
+ def workload_max_run_duration(self) -> 'str':
710
+ """Gets the workload_max_run_duration of this OrgsIdBody. # noqa: E501
711
+
712
+
713
+ :return: The workload_max_run_duration of this OrgsIdBody. # noqa: E501
714
+ :rtype: str
715
+ """
716
+ return self._workload_max_run_duration
717
+
718
+ @workload_max_run_duration.setter
719
+ def workload_max_run_duration(self, workload_max_run_duration: 'str'):
720
+ """Sets the workload_max_run_duration of this OrgsIdBody.
721
+
722
+
723
+ :param workload_max_run_duration: The workload_max_run_duration of this OrgsIdBody. # noqa: E501
724
+ :type: str
725
+ """
726
+
727
+ self._workload_max_run_duration = workload_max_run_duration
728
+
677
729
  def to_dict(self) -> dict:
678
730
  """Returns the model properties as a dict"""
679
731
  result = {}
@@ -46,6 +46,7 @@ class Update(object):
46
46
  'gcp': 'V1GcpDataConnection',
47
47
  'gcs_folder': 'V1GCSFolderDataConnection',
48
48
  'name': 'str',
49
+ 'r2': 'V1R2DataConnection',
49
50
  'run_cmds': 'list[str]',
50
51
  's3_folder': 'V1S3FolderDataConnection',
51
52
  'snowflake': 'V1SnowflakeDataConnection',
@@ -58,19 +59,21 @@ class Update(object):
58
59
  'gcp': 'gcp',
59
60
  'gcs_folder': 'gcsFolder',
60
61
  'name': 'name',
62
+ 'r2': 'r2',
61
63
  'run_cmds': 'runCmds',
62
64
  's3_folder': 's3Folder',
63
65
  'snowflake': 'snowflake',
64
66
  'writable': 'writable'
65
67
  }
66
68
 
67
- def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, name: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None): # noqa: E501
69
+ def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, name: 'str' =None, r2: 'V1R2DataConnection' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None): # noqa: E501
68
70
  """Update - a model defined in Swagger""" # noqa: E501
69
71
  self._aws = None
70
72
  self._efs = None
71
73
  self._gcp = None
72
74
  self._gcs_folder = None
73
75
  self._name = None
76
+ self._r2 = None
74
77
  self._run_cmds = None
75
78
  self._s3_folder = None
76
79
  self._snowflake = None
@@ -86,6 +89,8 @@ class Update(object):
86
89
  self.gcs_folder = gcs_folder
87
90
  if name is not None:
88
91
  self.name = name
92
+ if r2 is not None:
93
+ self.r2 = r2
89
94
  if run_cmds is not None:
90
95
  self.run_cmds = run_cmds
91
96
  if s3_folder is not None:
@@ -200,6 +205,27 @@ class Update(object):
200
205
 
201
206
  self._name = name
202
207
 
208
+ @property
209
+ def r2(self) -> 'V1R2DataConnection':
210
+ """Gets the r2 of this Update. # noqa: E501
211
+
212
+
213
+ :return: The r2 of this Update. # noqa: E501
214
+ :rtype: V1R2DataConnection
215
+ """
216
+ return self._r2
217
+
218
+ @r2.setter
219
+ def r2(self, r2: 'V1R2DataConnection'):
220
+ """Sets the r2 of this Update.
221
+
222
+
223
+ :param r2: The r2 of this Update. # noqa: E501
224
+ :type: V1R2DataConnection
225
+ """
226
+
227
+ self._r2 = r2
228
+
203
229
  @property
204
230
  def run_cmds(self) -> 'list[str]':
205
231
  """Gets the run_cmds of this Update. # noqa: E501
@@ -52,6 +52,7 @@ class V1CloudSpace(object):
52
52
  'display_name': 'str',
53
53
  'engagement_counts': 'dict(str, str)',
54
54
  'env': 'list[V1EnvVar]',
55
+ 'environment_template_id': 'str',
55
56
  'featured': 'bool',
56
57
  'hide_files': 'bool',
57
58
  'id': 'str',
@@ -64,6 +65,7 @@ class V1CloudSpace(object):
64
65
  'license_url': 'str',
65
66
  'lock_out': 'bool',
66
67
  'machine_image_version': 'str',
68
+ 'max_run_duration': 'str',
67
69
  'message': 'str',
68
70
  'multi_user_edit': 'bool',
69
71
  'name': 'str',
@@ -103,6 +105,7 @@ class V1CloudSpace(object):
103
105
  'display_name': 'displayName',
104
106
  'engagement_counts': 'engagementCounts',
105
107
  'env': 'env',
108
+ 'environment_template_id': 'environmentTemplateId',
106
109
  'featured': 'featured',
107
110
  'hide_files': 'hideFiles',
108
111
  'id': 'id',
@@ -115,6 +118,7 @@ class V1CloudSpace(object):
115
118
  'license_url': 'licenseUrl',
116
119
  'lock_out': 'lockOut',
117
120
  'machine_image_version': 'machineImageVersion',
121
+ 'max_run_duration': 'maxRunDuration',
118
122
  'message': 'message',
119
123
  'multi_user_edit': 'multiUserEdit',
120
124
  'name': 'name',
@@ -142,7 +146,7 @@ class V1CloudSpace(object):
142
146
  'web_path': 'webPath'
143
147
  }
144
148
 
145
- def __init__(self, can_download_source_code: 'bool' =None, cluster_id: 'str' =None, code_config: 'V1CloudSpaceInstanceConfig' =None, code_status: 'V1GetCloudSpaceInstanceStatusResponse' =None, code_url: 'str' =None, created_at: 'datetime' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, description: 'str' =None, display_name: 'str' =None, engagement_counts: 'dict(str, str)' =None, env: 'list[V1EnvVar]' =None, featured: 'bool' =None, hide_files: 'bool' =None, id: 'str' =None, image_status: 'V1ImageState' =None, is_cloudspace_private: 'bool' =None, is_code_private: 'bool' =None, is_favorite: 'bool' =None, is_published: 'bool' =None, license: 'str' =None, license_url: 'str' =None, lock_out: 'bool' =None, machine_image_version: 'str' =None, message: 'str' =None, multi_user_edit: 'bool' =None, name: 'str' =None, number_of_files: 'str' =None, operating_cost: 'str' =None, paper_authors: 'str' =None, paper_org: 'str' =None, paper_org_avatar_url: 'str' =None, paper_url: 'str' =None, project_id: 'str' =None, publications: 'list[V1CloudSpaceVersionPublication]' =None, published_at: 'datetime' =None, published_to_org_at: 'datetime' =None, state: 'V1CloudSpaceState' =None, sync_duration: 'str' =None, sync_percentage: 'str' =None, tags: 'list[V1ResourceTag]' =None, thumbnail_file_type: 'str' =None, thumbnail_updated_at: 'datetime' =None, thumbnail_url: 'str' =None, total_size_bytes: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None, user_metadata: 'str' =None, web_path: 'str' =None): # noqa: E501
149
+ def __init__(self, can_download_source_code: 'bool' =None, cluster_id: 'str' =None, code_config: 'V1CloudSpaceInstanceConfig' =None, code_status: 'V1GetCloudSpaceInstanceStatusResponse' =None, code_url: 'str' =None, created_at: 'datetime' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, description: 'str' =None, display_name: 'str' =None, engagement_counts: 'dict(str, str)' =None, env: 'list[V1EnvVar]' =None, environment_template_id: 'str' =None, featured: 'bool' =None, hide_files: 'bool' =None, id: 'str' =None, image_status: 'V1ImageState' =None, is_cloudspace_private: 'bool' =None, is_code_private: 'bool' =None, is_favorite: 'bool' =None, is_published: 'bool' =None, license: 'str' =None, license_url: 'str' =None, lock_out: 'bool' =None, machine_image_version: 'str' =None, max_run_duration: 'str' =None, message: 'str' =None, multi_user_edit: 'bool' =None, name: 'str' =None, number_of_files: 'str' =None, operating_cost: 'str' =None, paper_authors: 'str' =None, paper_org: 'str' =None, paper_org_avatar_url: 'str' =None, paper_url: 'str' =None, project_id: 'str' =None, publications: 'list[V1CloudSpaceVersionPublication]' =None, published_at: 'datetime' =None, published_to_org_at: 'datetime' =None, state: 'V1CloudSpaceState' =None, sync_duration: 'str' =None, sync_percentage: 'str' =None, tags: 'list[V1ResourceTag]' =None, thumbnail_file_type: 'str' =None, thumbnail_updated_at: 'datetime' =None, thumbnail_url: 'str' =None, total_size_bytes: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None, user_metadata: 'str' =None, web_path: 'str' =None): # noqa: E501
146
150
  """V1CloudSpace - a model defined in Swagger""" # noqa: E501
147
151
  self._can_download_source_code = None
148
152
  self._cluster_id = None
@@ -155,6 +159,7 @@ class V1CloudSpace(object):
155
159
  self._display_name = None
156
160
  self._engagement_counts = None
157
161
  self._env = None
162
+ self._environment_template_id = None
158
163
  self._featured = None
159
164
  self._hide_files = None
160
165
  self._id = None
@@ -167,6 +172,7 @@ class V1CloudSpace(object):
167
172
  self._license_url = None
168
173
  self._lock_out = None
169
174
  self._machine_image_version = None
175
+ self._max_run_duration = None
170
176
  self._message = None
171
177
  self._multi_user_edit = None
172
178
  self._name = None
@@ -215,6 +221,8 @@ class V1CloudSpace(object):
215
221
  self.engagement_counts = engagement_counts
216
222
  if env is not None:
217
223
  self.env = env
224
+ if environment_template_id is not None:
225
+ self.environment_template_id = environment_template_id
218
226
  if featured is not None:
219
227
  self.featured = featured
220
228
  if hide_files is not None:
@@ -239,6 +247,8 @@ class V1CloudSpace(object):
239
247
  self.lock_out = lock_out
240
248
  if machine_image_version is not None:
241
249
  self.machine_image_version = machine_image_version
250
+ if max_run_duration is not None:
251
+ self.max_run_duration = max_run_duration
242
252
  if message is not None:
243
253
  self.message = message
244
254
  if multi_user_edit is not None:
@@ -521,6 +531,27 @@ class V1CloudSpace(object):
521
531
 
522
532
  self._env = env
523
533
 
534
+ @property
535
+ def environment_template_id(self) -> 'str':
536
+ """Gets the environment_template_id of this V1CloudSpace. # noqa: E501
537
+
538
+
539
+ :return: The environment_template_id of this V1CloudSpace. # noqa: E501
540
+ :rtype: str
541
+ """
542
+ return self._environment_template_id
543
+
544
+ @environment_template_id.setter
545
+ def environment_template_id(self, environment_template_id: 'str'):
546
+ """Sets the environment_template_id of this V1CloudSpace.
547
+
548
+
549
+ :param environment_template_id: The environment_template_id of this V1CloudSpace. # noqa: E501
550
+ :type: str
551
+ """
552
+
553
+ self._environment_template_id = environment_template_id
554
+
524
555
  @property
525
556
  def featured(self) -> 'bool':
526
557
  """Gets the featured of this V1CloudSpace. # noqa: E501
@@ -773,6 +804,27 @@ class V1CloudSpace(object):
773
804
 
774
805
  self._machine_image_version = machine_image_version
775
806
 
807
+ @property
808
+ def max_run_duration(self) -> 'str':
809
+ """Gets the max_run_duration of this V1CloudSpace. # noqa: E501
810
+
811
+
812
+ :return: The max_run_duration of this V1CloudSpace. # noqa: E501
813
+ :rtype: str
814
+ """
815
+ return self._max_run_duration
816
+
817
+ @max_run_duration.setter
818
+ def max_run_duration(self, max_run_duration: 'str'):
819
+ """Sets the max_run_duration of this V1CloudSpace.
820
+
821
+
822
+ :param max_run_duration: The max_run_duration of this V1CloudSpace. # noqa: E501
823
+ :type: str
824
+ """
825
+
826
+ self._max_run_duration = max_run_duration
827
+
776
828
  @property
777
829
  def message(self) -> 'str':
778
830
  """Gets the message of this V1CloudSpace. # noqa: E501