lightning-sdk 0.1.35__py3-none-any.whl → 0.1.36__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py

@@ -44,6 +44,7 @@ class V1GetModelFilesResponse(object):
         'filepaths': 'list[str]',
         'model_id': 'str',
         'project_id': 'str',
+        'size_bytes': 'str',
         'version': 'str'
     }
 
@@ -51,14 +52,16 @@ class V1GetModelFilesResponse(object):
         'filepaths': 'filepaths',
         'model_id': 'modelId',
         'project_id': 'projectId',
+        'size_bytes': 'sizeBytes',
         'version': 'version'
     }
 
-    def __init__(self, filepaths: 'list[str]' =None, model_id: 'str' =None, project_id: 'str' =None, version: 'str' =None):  # noqa: E501
+    def __init__(self, filepaths: 'list[str]' =None, model_id: 'str' =None, project_id: 'str' =None, size_bytes: 'str' =None, version: 'str' =None):  # noqa: E501
         """V1GetModelFilesResponse - a model defined in Swagger"""  # noqa: E501
         self._filepaths = None
         self._model_id = None
         self._project_id = None
+        self._size_bytes = None
         self._version = None
         self.discriminator = None
         if filepaths is not None:
@@ -67,6 +70,8 @@ class V1GetModelFilesResponse(object):
             self.model_id = model_id
         if project_id is not None:
             self.project_id = project_id
+        if size_bytes is not None:
+            self.size_bytes = size_bytes
         if version is not None:
             self.version = version
 
@@ -133,6 +138,27 @@ class V1GetModelFilesResponse(object):
 
         self._project_id = project_id
 
+    @property
+    def size_bytes(self) -> 'str':
+        """Gets the size_bytes of this V1GetModelFilesResponse.  # noqa: E501
+
+
+        :return: The size_bytes of this V1GetModelFilesResponse.  # noqa: E501
+        :rtype: str
+        """
+        return self._size_bytes
+
+    @size_bytes.setter
+    def size_bytes(self, size_bytes: 'str'):
+        """Sets the size_bytes of this V1GetModelFilesResponse.
+
+
+        :param size_bytes: The size_bytes of this V1GetModelFilesResponse.  # noqa: E501
+        :type: str
+        """
+
+        self._size_bytes = size_bytes
+
     @property
     def version(self) -> 'str':
         """Gets the version of this V1GetModelFilesResponse.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py

@@ -45,6 +45,7 @@ class V1GetUserResponse(object):
         'api_key': 'str',
         'country': 'str',
         'email': 'str',
+        'experimentation_id': 'str',
         'features': 'V1UserFeatures',
         'first_name': 'str',
         'general_audience_mode': 'bool',
@@ -80,6 +81,7 @@ class V1GetUserResponse(object):
         'api_key': 'apiKey',
         'country': 'country',
         'email': 'email',
+        'experimentation_id': 'experimentationId',
         'features': 'features',
         'first_name': 'firstName',
         'general_audience_mode': 'generalAudienceMode',
@@ -110,12 +112,13 @@ class V1GetUserResponse(object):
         'website': 'website'
     }
 
-    def __init__(self, agree_to_terms_and_conditions: 'bool' =None, api_key: 'str' =None, country: 'str' =None, email: 'str' =None, features: 'V1UserFeatures' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, id: 'str' =None, internal_docs_admin: 'bool' =None, is_internal: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, organizations: 'list[V1Organization]' =None, phone_number: 'str' =None, picture_url: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, sb: 'bool' =None, status: 'Externalv1UserStatus' =None, storage_bytes: 'str' =None, user_metadata: 'str' =None, username: 'str' =None, waitlisted: 'bool' =None, website: 'str' =None):  # noqa: E501
+    def __init__(self, agree_to_terms_and_conditions: 'bool' =None, api_key: 'str' =None, country: 'str' =None, email: 'str' =None, experimentation_id: 'str' =None, features: 'V1UserFeatures' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, id: 'str' =None, internal_docs_admin: 'bool' =None, is_internal: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, organizations: 'list[V1Organization]' =None, phone_number: 'str' =None, picture_url: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, sb: 'bool' =None, status: 'Externalv1UserStatus' =None, storage_bytes: 'str' =None, user_metadata: 'str' =None, username: 'str' =None, waitlisted: 'bool' =None, website: 'str' =None):  # noqa: E501
         """V1GetUserResponse - a model defined in Swagger"""  # noqa: E501
         self._agree_to_terms_and_conditions = None
         self._api_key = None
         self._country = None
         self._email = None
+        self._experimentation_id = None
         self._features = None
         self._first_name = None
         self._general_audience_mode = None
@@ -153,6 +156,8 @@ class V1GetUserResponse(object):
             self.country = country
         if email is not None:
             self.email = email
+        if experimentation_id is not None:
+            self.experimentation_id = experimentation_id
         if features is not None:
             self.features = features
         if first_name is not None:
@@ -294,6 +299,27 @@ class V1GetUserResponse(object):
 
         self._email = email
 
+    @property
+    def experimentation_id(self) -> 'str':
+        """Gets the experimentation_id of this V1GetUserResponse.  # noqa: E501
+
+
+        :return: The experimentation_id of this V1GetUserResponse.  # noqa: E501
+        :rtype: str
+        """
+        return self._experimentation_id
+
+    @experimentation_id.setter
+    def experimentation_id(self, experimentation_id: 'str'):
+        """Sets the experimentation_id of this V1GetUserResponse.
+
+
+        :param experimentation_id: The experimentation_id of this V1GetUserResponse.  # noqa: E501
+        :type: str
+        """
+
+        self._experimentation_id = experimentation_id
+
     @property
     def features(self) -> 'V1UserFeatures':
         """Gets the features of this V1GetUserResponse.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py

@@ -43,6 +43,7 @@ class V1MagicLinkLoginRequest(object):
     swagger_types = {
         'collab_session_id': 'str',
         'email': 'str',
+        'experimentation_id': 'str',
         'invite_code': 'str',
         'redirect_url': 'str',
         'referrer_params': 'str'
@@ -51,15 +52,17 @@ class V1MagicLinkLoginRequest(object):
     attribute_map = {
         'collab_session_id': 'collabSessionId',
         'email': 'email',
+        'experimentation_id': 'experimentationId',
         'invite_code': 'inviteCode',
         'redirect_url': 'redirectUrl',
         'referrer_params': 'referrerParams'
     }
 
-    def __init__(self, collab_session_id: 'str' =None, email: 'str' =None, invite_code: 'str' =None, redirect_url: 'str' =None, referrer_params: 'str' =None):  # noqa: E501
+    def __init__(self, collab_session_id: 'str' =None, email: 'str' =None, experimentation_id: 'str' =None, invite_code: 'str' =None, redirect_url: 'str' =None, referrer_params: 'str' =None):  # noqa: E501
         """V1MagicLinkLoginRequest - a model defined in Swagger"""  # noqa: E501
         self._collab_session_id = None
         self._email = None
+        self._experimentation_id = None
         self._invite_code = None
         self._redirect_url = None
         self._referrer_params = None
@@ -68,6 +71,8 @@ class V1MagicLinkLoginRequest(object):
             self.collab_session_id = collab_session_id
         if email is not None:
             self.email = email
+        if experimentation_id is not None:
+            self.experimentation_id = experimentation_id
         if invite_code is not None:
             self.invite_code = invite_code
         if redirect_url is not None:
@@ -117,6 +122,27 @@ class V1MagicLinkLoginRequest(object):
 
         self._email = email
 
+    @property
+    def experimentation_id(self) -> 'str':
+        """Gets the experimentation_id of this V1MagicLinkLoginRequest.  # noqa: E501
+
+
+        :return: The experimentation_id of this V1MagicLinkLoginRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._experimentation_id
+
+    @experimentation_id.setter
+    def experimentation_id(self, experimentation_id: 'str'):
+        """Sets the experimentation_id of this V1MagicLinkLoginRequest.
+
+
+        :param experimentation_id: The experimentation_id of this V1MagicLinkLoginRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._experimentation_id = experimentation_id
+
     @property
     def invite_code(self) -> 'str':
         """Gets the invite_code of this V1MagicLinkLoginRequest.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_metric_value.py

@@ -42,29 +42,24 @@ class V1MetricValue(object):
     """
     swagger_types = {
         'created_at': 'datetime',
-        'internal_step': 'str',
         'step': 'str',
         'value': 'float'
     }
 
     attribute_map = {
         'created_at': 'createdAt',
-        'internal_step': 'internalStep',
         'step': 'step',
         'value': 'value'
     }
 
-    def __init__(self, created_at: 'datetime' =None, internal_step: 'str' =None, step: 'str' =None, value: 'float' =None):  # noqa: E501
+    def __init__(self, created_at: 'datetime' =None, step: 'str' =None, value: 'float' =None):  # noqa: E501
         """V1MetricValue - a model defined in Swagger"""  # noqa: E501
         self._created_at = None
-        self._internal_step = None
         self._step = None
         self._value = None
         self.discriminator = None
         if created_at is not None:
             self.created_at = created_at
-        if internal_step is not None:
-            self.internal_step = internal_step
         if step is not None:
             self.step = step
         if value is not None:
@@ -91,27 +86,6 @@ class V1MetricValue(object):
 
         self._created_at = created_at
 
-    @property
-    def internal_step(self) -> 'str':
-        """Gets the internal_step of this V1MetricValue.  # noqa: E501
-
-
-        :return: The internal_step of this V1MetricValue.  # noqa: E501
-        :rtype: str
-        """
-        return self._internal_step
-
-    @internal_step.setter
-    def internal_step(self, internal_step: 'str'):
-        """Sets the internal_step of this V1MetricValue.
-
-
-        :param internal_step: The internal_step of this V1MetricValue.  # noqa: E501
-        :type: str
-        """
-
-        self._internal_step = internal_step
-
     @property
     def step(self) -> 'str':
         """Gets the step of this V1MetricValue.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_metrics.py

@@ -41,25 +41,51 @@ class V1Metrics(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'internal_start_step': 'int',
         'name': 'str',
         'values': 'list[V1MetricValue]'
     }
 
     attribute_map = {
+        'internal_start_step': 'internalStartStep',
         'name': 'name',
         'values': 'values'
     }
 
-    def __init__(self, name: 'str' =None, values: 'list[V1MetricValue]' =None):  # noqa: E501
+    def __init__(self, internal_start_step: 'int' =None, name: 'str' =None, values: 'list[V1MetricValue]' =None):  # noqa: E501
         """V1Metrics - a model defined in Swagger"""  # noqa: E501
+        self._internal_start_step = None
         self._name = None
         self._values = None
         self.discriminator = None
+        if internal_start_step is not None:
+            self.internal_start_step = internal_start_step
         if name is not None:
             self.name = name
         if values is not None:
             self.values = values
 
+    @property
+    def internal_start_step(self) -> 'int':
+        """Gets the internal_start_step of this V1Metrics.  # noqa: E501
+
+
+        :return: The internal_start_step of this V1Metrics.  # noqa: E501
+        :rtype: int
+        """
+        return self._internal_start_step
+
+    @internal_start_step.setter
+    def internal_start_step(self, internal_start_step: 'int'):
+        """Sets the internal_start_step of this V1Metrics.
+
+
+        :param internal_start_step: The internal_start_step of this V1Metrics.  # noqa: E501
+        :type: int
+        """
+
+        self._internal_start_step = internal_start_step
+
     @property
     def name(self) -> 'str':
         """Gets the name of this V1Metrics.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py

@@ -46,6 +46,7 @@ class V1UpdateUserRequest(object):
         'completed_project_onboarding': 'bool',
         'country': 'str',
         'email': 'str',
+        'experimentation_id': 'str',
         'first_name': 'str',
         'general_audience_mode': 'bool',
         'last_name': 'str',
@@ -71,6 +72,7 @@ class V1UpdateUserRequest(object):
         'completed_project_onboarding': 'completedProjectOnboarding',
         'country': 'country',
         'email': 'email',
+        'experimentation_id': 'experimentationId',
         'first_name': 'firstName',
         'general_audience_mode': 'generalAudienceMode',
         'last_name': 'lastName',
@@ -90,13 +92,14 @@ class V1UpdateUserRequest(object):
         'website': 'website'
     }
 
-    def __init__(self, agree_to_terms_and_conditions: 'bool' =None, complete_sign_up: 'bool' =None, completed_project_onboarding: 'bool' =None, country: 'str' =None, email: 'str' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, user_metadata: 'str' =None, username: 'str' =None, website: 'str' =None):  # noqa: E501
+    def __init__(self, agree_to_terms_and_conditions: 'bool' =None, complete_sign_up: 'bool' =None, completed_project_onboarding: 'bool' =None, country: 'str' =None, email: 'str' =None, experimentation_id: 'str' =None, first_name: 'str' =None, general_audience_mode: 'bool' =None, last_name: 'str' =None, non_developer_mode: 'bool' =None, opted_in_marketing_emails: 'bool' =None, organization: 'str' =None, preferred_color_scheme: 'str' =None, preferred_ide: 'str' =None, preferred_shell: 'str' =None, preferred_vscode_marketplace: 'str' =None, role: 'str' =None, saw_create_first_project_dialog: 'bool' =None, saw_forums_login_merge_dialog: 'bool' =None, saw_free_credits_notification: 'bool' =None, user_metadata: 'str' =None, username: 'str' =None, website: 'str' =None):  # noqa: E501
         """V1UpdateUserRequest - a model defined in Swagger"""  # noqa: E501
         self._agree_to_terms_and_conditions = None
         self._complete_sign_up = None
         self._completed_project_onboarding = None
         self._country = None
         self._email = None
+        self._experimentation_id = None
         self._first_name = None
         self._general_audience_mode = None
         self._last_name = None
@@ -125,6 +128,8 @@ class V1UpdateUserRequest(object):
             self.country = country
         if email is not None:
             self.email = email
+        if experimentation_id is not None:
+            self.experimentation_id = experimentation_id
         if first_name is not None:
             self.first_name = first_name
         if general_audience_mode is not None:
@@ -265,6 +270,27 @@ class V1UpdateUserRequest(object):
 
         self._email = email
 
+    @property
+    def experimentation_id(self) -> 'str':
+        """Gets the experimentation_id of this V1UpdateUserRequest.  # noqa: E501
+
+
+        :return: The experimentation_id of this V1UpdateUserRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._experimentation_id
+
+    @experimentation_id.setter
+    def experimentation_id(self, experimentation_id: 'str'):
+        """Sets the experimentation_id of this V1UpdateUserRequest.
+
+
+        :param experimentation_id: The experimentation_id of this V1UpdateUserRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._experimentation_id = experimentation_id
+
     @property
     def first_name(self) -> 'str':
         """Gets the first_name of this V1UpdateUserRequest.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -76,6 +76,7 @@ class V1UserFeatures(object):
         'jobs_init': 'bool',
         'jobs_v2': 'bool',
         'landing_studios': 'bool',
+        'lightning_registry': 'bool',
         'lit_logger': 'bool',
         'lit_logger_storage_v2': 'bool',
         'mmt_fault_tolerance': 'bool',
@@ -154,6 +155,7 @@ class V1UserFeatures(object):
         'jobs_init': 'jobsInit',
         'jobs_v2': 'jobsV2',
         'landing_studios': 'landingStudios',
+        'lightning_registry': 'lightningRegistry',
         'lit_logger': 'litLogger',
         'lit_logger_storage_v2': 'litLoggerStorageV2',
         'mmt_fault_tolerance': 'mmtFaultTolerance',
@@ -196,7 +198,7 @@ class V1UserFeatures(object):
         'writable_data_connections': 'writableDataConnections'
     }
 
-    def __init__(self, advanced_deployment_autoscaling: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: '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: '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, docs_agent: 'bool' =None, drive_v2: '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, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: '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, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, teamspace_storage_tab: 'bool' =None, use_rclone_mounts_only: 'bool' =None, writable_data_connections: 'bool' =None):  # noqa: E501
+    def __init__(self, advanced_deployment_autoscaling: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: '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: '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, docs_agent: 'bool' =None, drive_v2: '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, 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, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, teamspace_storage_tab: 'bool' =None, use_rclone_mounts_only: 'bool' =None, writable_data_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._advanced_deployment_autoscaling = None
         self._affiliate_links = None
@@ -233,6 +235,7 @@ class V1UserFeatures(object):
         self._jobs_init = None
         self._jobs_v2 = None
         self._landing_studios = None
+        self._lightning_registry = None
         self._lit_logger = None
         self._lit_logger_storage_v2 = None
         self._mmt_fault_tolerance = None
@@ -344,6 +347,8 @@ class V1UserFeatures(object):
             self.jobs_v2 = jobs_v2
         if landing_studios is not None:
             self.landing_studios = landing_studios
+        if lightning_registry is not None:
+            self.lightning_registry = lightning_registry
         if lit_logger is not None:
             self.lit_logger = lit_logger
         if lit_logger_storage_v2 is not None:
@@ -1160,6 +1165,27 @@ class V1UserFeatures(object):
 
         self._landing_studios = landing_studios
 
+    @property
+    def lightning_registry(self) -> 'bool':
+        """Gets the lightning_registry of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The lightning_registry of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._lightning_registry
+
+    @lightning_registry.setter
+    def lightning_registry(self, lightning_registry: 'bool'):
+        """Sets the lightning_registry of this V1UserFeatures.
+
+
+        :param lightning_registry: The lightning_registry of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._lightning_registry = lightning_registry
+
     @property
     def lit_logger(self) -> 'bool':
         """Gets the lit_logger of this V1UserFeatures.  # noqa: E501

lightning_sdk/lightning_cloud/rest_client.py

@@ -4,7 +4,6 @@ import time
 from functools import wraps
 from typing import Callable, Optional, Any
 
-import click
 import urllib3
 from lightning_sdk.lightning_cloud import env
 from lightning_sdk.lightning_cloud.login import Auth
@@ -28,7 +27,6 @@ from lightning_sdk.lightning_cloud.openapi import (
     EndpointServiceApi,
     SlurmJobsUserServiceApi,
     LitLoggerServiceApi,
-    AssistantsServiceApi,
     JobsServiceApi,
     AssistantsServiceApi,
     StorageServiceApi,
@@ -199,8 +197,7 @@ def request_auth_warning_wrapper(func):
             return response
         except ApiException as err:
             if err.status == 401:
-                raise click.ClickException(
-                    "Authentication failed. Please run `lightning login`.")
+                raise ConnectionError("Authentication failed. Please run `lightning login`.")
             raise err
 
     return wrap

lightning_sdk/lightning_cloud/source_code/tar.py

@@ -5,8 +5,6 @@ import subprocess
 import tarfile
 from typing import Optional, Tuple
 
-import click
-
 MAX_SPLIT_COUNT = 999
 
 
@@ -85,7 +83,7 @@ def get_split_size(total_size: int,
         1 << 31
     )  # max size per part limited by Requests or urllib as shown in ref above
     if total_size > max_size:
-        raise click.ClickException(
+        raise RuntimeError(
             f"The size of the datastore to be uploaded is bigger than our {max_size/(1 << 40):.2f} TBytes limit"
         )
 

{lightning_sdk-0.1.35.dist-info → lightning_sdk-0.1.36.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 0.1.35
+Version: 0.1.36
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License
@@ -32,7 +32,6 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: backoff
-Requires-Dist: click
 Requires-Dist: fastapi
 Requires-Dist: pyjwt
 Requires-Dist: python-multipart