lightning-sdk 0.1.41__py3-none-any.whl → 0.1.43__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -47,7 +47,6 @@ class V1UserFeatures(object):
         'ai_hub_monetization': 'bool',
         'auto_fast_load': 'bool',
         'auto_join_orgs': 'bool',
-        'aws_trainium': 'bool',
         'b2c_experience': 'bool',
         'cap_add': 'list[str]',
         'cap_drop': 'list[str]',
@@ -57,7 +56,6 @@ class V1UserFeatures(object):
         'code_tab': 'bool',
         'collab_screen_sharing': 'bool',
         'cost_attribution_settings': 'bool',
-        'crypto_monitoring': 'bool',
         'custom_app_domain': 'bool',
         'custom_instance_types': 'bool',
         'default_one_cluster': 'bool',
@@ -85,6 +83,7 @@ class V1UserFeatures(object):
         'lit_logger_storage_v2': 'bool',
         'mmt_fault_tolerance': 'bool',
         'mmt_strategy_selector': 'bool',
+        'mmt_v2': 'bool',
         'model_store': 'bool',
         'multiple_deployment_versions': 'bool',
         'multiple_studio_versions': 'bool',
@@ -132,7 +131,6 @@ class V1UserFeatures(object):
         'ai_hub_monetization': 'aiHubMonetization',
         'auto_fast_load': 'autoFastLoad',
         'auto_join_orgs': 'autoJoinOrgs',
-        'aws_trainium': 'awsTrainium',
         'b2c_experience': 'b2cExperience',
         'cap_add': 'capAdd',
         'cap_drop': 'capDrop',
@@ -142,7 +140,6 @@ class V1UserFeatures(object):
         'code_tab': 'codeTab',
         'collab_screen_sharing': 'collabScreenSharing',
         'cost_attribution_settings': 'costAttributionSettings',
-        'crypto_monitoring': 'cryptoMonitoring',
         'custom_app_domain': 'customAppDomain',
         'custom_instance_types': 'customInstanceTypes',
         'default_one_cluster': 'defaultOneCluster',
@@ -170,6 +167,7 @@ class V1UserFeatures(object):
         'lit_logger_storage_v2': 'litLoggerStorageV2',
         'mmt_fault_tolerance': 'mmtFaultTolerance',
         'mmt_strategy_selector': 'mmtStrategySelector',
+        'mmt_v2': 'mmtV2',
         'model_store': 'modelStore',
         'multiple_deployment_versions': 'multipleDeploymentVersions',
         'multiple_studio_versions': 'multipleStudioVersions',
@@ -210,7 +208,7 @@ class V1UserFeatures(object):
         'writable_data_connections': 'writableDataConnections'
     }
 
-    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
+    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
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._advanced_deployment_autoscaling = None
         self._affiliate_links = None
@@ -218,7 +216,6 @@ class V1UserFeatures(object):
         self._ai_hub_monetization = None
         self._auto_fast_load = None
         self._auto_join_orgs = None
-        self._aws_trainium = None
         self._b2c_experience = None
         self._cap_add = None
         self._cap_drop = None
@@ -228,7 +225,6 @@ class V1UserFeatures(object):
         self._code_tab = None
         self._collab_screen_sharing = None
         self._cost_attribution_settings = None
-        self._crypto_monitoring = None
         self._custom_app_domain = None
         self._custom_instance_types = None
         self._default_one_cluster = None
@@ -256,6 +252,7 @@ class V1UserFeatures(object):
         self._lit_logger_storage_v2 = None
         self._mmt_fault_tolerance = None
         self._mmt_strategy_selector = None
+        self._mmt_v2 = None
         self._model_store = None
         self._multiple_deployment_versions = None
         self._multiple_studio_versions = None
@@ -307,8 +304,6 @@ class V1UserFeatures(object):
             self.auto_fast_load = auto_fast_load
         if auto_join_orgs is not None:
             self.auto_join_orgs = auto_join_orgs
-        if aws_trainium is not None:
-            self.aws_trainium = aws_trainium
         if b2c_experience is not None:
             self.b2c_experience = b2c_experience
         if cap_add is not None:
@@ -327,8 +322,6 @@ class V1UserFeatures(object):
             self.collab_screen_sharing = collab_screen_sharing
         if cost_attribution_settings is not None:
             self.cost_attribution_settings = cost_attribution_settings
-        if crypto_monitoring is not None:
-            self.crypto_monitoring = crypto_monitoring
         if custom_app_domain is not None:
             self.custom_app_domain = custom_app_domain
         if custom_instance_types is not None:
@@ -383,6 +376,8 @@ class V1UserFeatures(object):
             self.mmt_fault_tolerance = mmt_fault_tolerance
         if mmt_strategy_selector is not None:
             self.mmt_strategy_selector = mmt_strategy_selector
+        if mmt_v2 is not None:
+            self.mmt_v2 = mmt_v2
         if model_store is not None:
             self.model_store = model_store
         if multiple_deployment_versions is not None:
@@ -586,27 +581,6 @@ class V1UserFeatures(object):
 
         self._auto_join_orgs = auto_join_orgs
 
-    @property
-    def aws_trainium(self) -> 'bool':
-        """Gets the aws_trainium of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The aws_trainium of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._aws_trainium
-
-    @aws_trainium.setter
-    def aws_trainium(self, aws_trainium: 'bool'):
-        """Sets the aws_trainium of this V1UserFeatures.
-
-
-        :param aws_trainium: The aws_trainium of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._aws_trainium = aws_trainium
-
     @property
     def b2c_experience(self) -> 'bool':
         """Gets the b2c_experience of this V1UserFeatures.  # noqa: E501
@@ -796,27 +770,6 @@ class V1UserFeatures(object):
 
         self._cost_attribution_settings = cost_attribution_settings
 
-    @property
-    def crypto_monitoring(self) -> 'bool':
-        """Gets the crypto_monitoring of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The crypto_monitoring of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._crypto_monitoring
-
-    @crypto_monitoring.setter
-    def crypto_monitoring(self, crypto_monitoring: 'bool'):
-        """Sets the crypto_monitoring of this V1UserFeatures.
-
-
-        :param crypto_monitoring: The crypto_monitoring of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._crypto_monitoring = crypto_monitoring
-
     @property
     def custom_app_domain(self) -> 'bool':
         """Gets the custom_app_domain of this V1UserFeatures.  # noqa: E501
@@ -1384,6 +1337,27 @@ class V1UserFeatures(object):
 
         self._mmt_strategy_selector = mmt_strategy_selector
 
+    @property
+    def mmt_v2(self) -> 'bool':
+        """Gets the mmt_v2 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The mmt_v2 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._mmt_v2
+
+    @mmt_v2.setter
+    def mmt_v2(self, mmt_v2: 'bool'):
+        """Sets the mmt_v2 of this V1UserFeatures.
+
+
+        :param mmt_v2: The mmt_v2 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._mmt_v2 = mmt_v2
+
     @property
     def model_store(self) -> 'bool':
         """Gets the model_store of this V1UserFeatures.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_request.py

@@ -0,0 +1,149 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1ValidateDeploymentImageRequest(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+        'image': 'str',
+        'secret': 'str'
+    }
+
+    attribute_map = {
+        'image': 'image',
+        'secret': 'secret'
+    }
+
+    def __init__(self, image: 'str' =None, secret: 'str' =None):  # noqa: E501
+        """V1ValidateDeploymentImageRequest - a model defined in Swagger"""  # noqa: E501
+        self._image = None
+        self._secret = None
+        self.discriminator = None
+        if image is not None:
+            self.image = image
+        if secret is not None:
+            self.secret = secret
+
+    @property
+    def image(self) -> 'str':
+        """Gets the image of this V1ValidateDeploymentImageRequest.  # noqa: E501
+
+
+        :return: The image of this V1ValidateDeploymentImageRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._image
+
+    @image.setter
+    def image(self, image: 'str'):
+        """Sets the image of this V1ValidateDeploymentImageRequest.
+
+
+        :param image: The image of this V1ValidateDeploymentImageRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._image = image
+
+    @property
+    def secret(self) -> 'str':
+        """Gets the secret of this V1ValidateDeploymentImageRequest.  # noqa: E501
+
+
+        :return: The secret of this V1ValidateDeploymentImageRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._secret
+
+    @secret.setter
+    def secret(self, secret: 'str'):
+        """Sets the secret of this V1ValidateDeploymentImageRequest.
+
+
+        :param secret: The secret of this V1ValidateDeploymentImageRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._secret = secret
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1ValidateDeploymentImageRequest, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1ValidateDeploymentImageRequest') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ValidateDeploymentImageRequest):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ValidateDeploymentImageRequest') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_response.py

@@ -0,0 +1,97 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1ValidateDeploymentImageResponse(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+    }
+
+    attribute_map = {
+    }
+
+    def __init__(self):  # noqa: E501
+        """V1ValidateDeploymentImageResponse - a model defined in Swagger"""  # noqa: E501
+        self.discriminator = None
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1ValidateDeploymentImageResponse, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1ValidateDeploymentImageResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ValidateDeploymentImageResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ValidateDeploymentImageResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/rest_client.py

@@ -31,6 +31,7 @@ from lightning_sdk.lightning_cloud.openapi import (
     AssistantsServiceApi,
     StorageServiceApi,
     DeploymentTemplatesServiceApi,
+    ModelsStoreApi,
 )
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 from lightning_sdk.lightning_cloud.source_code.logs_socket_api import LightningLogsSocketAPI
@@ -89,6 +90,7 @@ class GridRestClient(
         AssistantsServiceApi,
         StorageServiceApi,
         DeploymentTemplatesServiceApi,
+        ModelsStoreApi,
 ):
 
     def __init__(self, api_client: Optional[ApiClient] = None):

lightning_sdk/mmt/__init__.py

@@ -0,0 +1,4 @@
+from lightning_sdk.mmt.base import MMTMachine
+from lightning_sdk.mmt.mmt import MMT
+
+__all__ = ["MMT", "MMTMachine"]