lightning-sdk 2025.12.9__py3-none-any.whl → 2025.12.17__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_refresh_container_registry_credentials_response.py

@@ -0,0 +1,123 @@
+# 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 V1RefreshContainerRegistryCredentialsResponse(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 = {
+        'refreshed_at': 'datetime'
+    }
+
+    attribute_map = {
+        'refreshed_at': 'refreshedAt'
+    }
+
+    def __init__(self, refreshed_at: 'datetime' =None):  # noqa: E501
+        """V1RefreshContainerRegistryCredentialsResponse - a model defined in Swagger"""  # noqa: E501
+        self._refreshed_at = None
+        self.discriminator = None
+        if refreshed_at is not None:
+            self.refreshed_at = refreshed_at
+
+    @property
+    def refreshed_at(self) -> 'datetime':
+        """Gets the refreshed_at of this V1RefreshContainerRegistryCredentialsResponse.  # noqa: E501
+
+
+        :return: The refreshed_at of this V1RefreshContainerRegistryCredentialsResponse.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._refreshed_at
+
+    @refreshed_at.setter
+    def refreshed_at(self, refreshed_at: 'datetime'):
+        """Sets the refreshed_at of this V1RefreshContainerRegistryCredentialsResponse.
+
+
+        :param refreshed_at: The refreshed_at of this V1RefreshContainerRegistryCredentialsResponse.  # noqa: E501
+        :type: datetime
+        """
+
+        self._refreshed_at = refreshed_at
+
+    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(V1RefreshContainerRegistryCredentialsResponse, 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: 'V1RefreshContainerRegistryCredentialsResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1RefreshContainerRegistryCredentialsResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1RefreshContainerRegistryCredentialsResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_search_user.py

@@ -46,6 +46,7 @@ class V1SearchUser(object):
         'email': 'str',
         'first_name': 'str',
         'id': 'str',
+        'internal': 'bool',
         'last_name': 'str',
         'organization': 'str',
         'picture_url': 'str',
@@ -61,6 +62,7 @@ class V1SearchUser(object):
         'email': 'email',
         'first_name': 'firstName',
         'id': 'id',
+        'internal': 'internal',
         'last_name': 'lastName',
         'organization': 'organization',
         'picture_url': 'pictureUrl',
@@ -70,13 +72,14 @@ class V1SearchUser(object):
         'website': 'website'
     }
 
-    def __init__(self, auth_provider: 'str' =None, country: 'str' =None, email: 'str' =None, first_name: 'str' =None, id: 'str' =None, last_name: 'str' =None, organization: 'str' =None, picture_url: 'str' =None, project_memberships: 'list[str]' =None, role: 'str' =None, username: 'str' =None, website: 'str' =None):  # noqa: E501
+    def __init__(self, auth_provider: 'str' =None, country: 'str' =None, email: 'str' =None, first_name: 'str' =None, id: 'str' =None, internal: 'bool' =None, last_name: 'str' =None, organization: 'str' =None, picture_url: 'str' =None, project_memberships: 'list[str]' =None, role: 'str' =None, username: 'str' =None, website: 'str' =None):  # noqa: E501
         """V1SearchUser - a model defined in Swagger"""  # noqa: E501
         self._auth_provider = None
         self._country = None
         self._email = None
         self._first_name = None
         self._id = None
+        self._internal = None
         self._last_name = None
         self._organization = None
         self._picture_url = None
@@ -95,6 +98,8 @@ class V1SearchUser(object):
             self.first_name = first_name
         if id is not None:
             self.id = id
+        if internal is not None:
+            self.internal = internal
         if last_name is not None:
             self.last_name = last_name
         if organization is not None:
@@ -215,6 +220,27 @@ class V1SearchUser(object):
 
         self._id = id
 
+    @property
+    def internal(self) -> 'bool':
+        """Gets the internal of this V1SearchUser.  # noqa: E501
+
+
+        :return: The internal of this V1SearchUser.  # noqa: E501
+        :rtype: bool
+        """
+        return self._internal
+
+    @internal.setter
+    def internal(self, internal: 'bool'):
+        """Sets the internal of this V1SearchUser.
+
+
+        :param internal: The internal of this V1SearchUser.  # noqa: E501
+        :type: bool
+        """
+
+        self._internal = internal
+
     @property
     def last_name(self) -> 'str':
         """Gets the last_name of this V1SearchUser.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -93,6 +93,9 @@ class V1UserFeatures(object):
         'f284': 'bool',
         'f285': 'bool',
         'f286': 'bool',
+        'f287': 'bool',
+        'f288': 'bool',
+        'f289': 'bool',
         'fair_share': 'bool',
         'featured_studios_admin': 'bool',
         'job_artifacts_v2': 'bool',
@@ -184,6 +187,9 @@ class V1UserFeatures(object):
         'f284': 'f284',
         'f285': 'f285',
         'f286': 'f286',
+        'f287': 'f287',
+        'f288': 'f288',
+        'f289': 'f289',
         'fair_share': 'fairShare',
         'featured_studios_admin': 'featuredStudiosAdmin',
         'job_artifacts_v2': 'jobArtifactsV2',
@@ -222,7 +228,7 @@ class V1UserFeatures(object):
         'weka': 'weka'
     }
 
-    def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, b2c_experience: 'bool' =None, byo_machine_type: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloudspace_schedules: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, f247: 'bool' =None, f250: 'bool' =None, f252: 'bool' =None, f253: 'bool' =None, f254: 'bool' =None, f258: 'bool' =None, f259: 'bool' =None, f265: 'bool' =None, f266: 'bool' =None, f268: 'bool' =None, f270: 'bool' =None, f271: 'bool' =None, f272: 'bool' =None, f273: 'bool' =None, f274: 'bool' =None, f275: 'bool' =None, f276: 'bool' =None, f278: 'bool' =None, f279: 'bool' =None, f280: 'bool' =None, f281: 'bool' =None, f282: 'bool' =None, f283: 'bool' =None, f284: 'bool' =None, f285: 'bool' =None, f286: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, persistent_disk: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_sweeps: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_version_visibility: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None):  # noqa: E501
+    def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, b2c_experience: 'bool' =None, byo_machine_type: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloudspace_schedules: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, f247: 'bool' =None, f250: 'bool' =None, f252: 'bool' =None, f253: 'bool' =None, f254: 'bool' =None, f258: 'bool' =None, f259: 'bool' =None, f265: 'bool' =None, f266: 'bool' =None, f268: 'bool' =None, f270: 'bool' =None, f271: 'bool' =None, f272: 'bool' =None, f273: 'bool' =None, f274: 'bool' =None, f275: 'bool' =None, f276: 'bool' =None, f278: 'bool' =None, f279: 'bool' =None, f280: 'bool' =None, f281: 'bool' =None, f282: 'bool' =None, f283: 'bool' =None, f284: 'bool' =None, f285: 'bool' =None, f286: 'bool' =None, f287: 'bool' =None, f288: 'bool' =None, f289: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, persistent_disk: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_sweeps: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_version_visibility: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._affiliate_links = None
         self._agents_v2 = None
@@ -276,6 +282,9 @@ class V1UserFeatures(object):
         self._f284 = None
         self._f285 = None
         self._f286 = None
+        self._f287 = None
+        self._f288 = None
+        self._f289 = None
         self._fair_share = None
         self._featured_studios_admin = None
         self._job_artifacts_v2 = None
@@ -417,6 +426,12 @@ class V1UserFeatures(object):
             self.f285 = f285
         if f286 is not None:
             self.f286 = f286
+        if f287 is not None:
+            self.f287 = f287
+        if f288 is not None:
+            self.f288 = f288
+        if f289 is not None:
+            self.f289 = f289
         if fair_share is not None:
             self.fair_share = fair_share
         if featured_studios_admin is not None:
@@ -1582,6 +1597,69 @@ class V1UserFeatures(object):
 
         self._f286 = f286
 
+    @property
+    def f287(self) -> 'bool':
+        """Gets the f287 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f287 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f287
+
+    @f287.setter
+    def f287(self, f287: 'bool'):
+        """Sets the f287 of this V1UserFeatures.
+
+
+        :param f287: The f287 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f287 = f287
+
+    @property
+    def f288(self) -> 'bool':
+        """Gets the f288 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f288 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f288
+
+    @f288.setter
+    def f288(self, f288: 'bool'):
+        """Sets the f288 of this V1UserFeatures.
+
+
+        :param f288: The f288 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f288 = f288
+
+    @property
+    def f289(self) -> 'bool':
+        """Gets the f289 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f289 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f289
+
+    @f289.setter
+    def f289(self, f289: 'bool'):
+        """Sets the f289 of this V1UserFeatures.
+
+
+        :param f289: The f289 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f289 = f289
+
     @property
     def fair_share(self) -> 'bool':
         """Gets the fair_share of this V1UserFeatures.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_validate_container_registry_response.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 V1ValidateContainerRegistryResponse(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 = {
+        'message': 'str',
+        'valid': 'bool'
+    }
+
+    attribute_map = {
+        'message': 'message',
+        'valid': 'valid'
+    }
+
+    def __init__(self, message: 'str' =None, valid: 'bool' =None):  # noqa: E501
+        """V1ValidateContainerRegistryResponse - a model defined in Swagger"""  # noqa: E501
+        self._message = None
+        self._valid = None
+        self.discriminator = None
+        if message is not None:
+            self.message = message
+        if valid is not None:
+            self.valid = valid
+
+    @property
+    def message(self) -> 'str':
+        """Gets the message of this V1ValidateContainerRegistryResponse.  # noqa: E501
+
+
+        :return: The message of this V1ValidateContainerRegistryResponse.  # noqa: E501
+        :rtype: str
+        """
+        return self._message
+
+    @message.setter
+    def message(self, message: 'str'):
+        """Sets the message of this V1ValidateContainerRegistryResponse.
+
+
+        :param message: The message of this V1ValidateContainerRegistryResponse.  # noqa: E501
+        :type: str
+        """
+
+        self._message = message
+
+    @property
+    def valid(self) -> 'bool':
+        """Gets the valid of this V1ValidateContainerRegistryResponse.  # noqa: E501
+
+
+        :return: The valid of this V1ValidateContainerRegistryResponse.  # noqa: E501
+        :rtype: bool
+        """
+        return self._valid
+
+    @valid.setter
+    def valid(self, valid: 'bool'):
+        """Sets the valid of this V1ValidateContainerRegistryResponse.
+
+
+        :param valid: The valid of this V1ValidateContainerRegistryResponse.  # noqa: E501
+        :type: bool
+        """
+
+        self._valid = valid
+
+    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(V1ValidateContainerRegistryResponse, 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: 'V1ValidateContainerRegistryResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ValidateContainerRegistryResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ValidateContainerRegistryResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/rest.py

@@ -49,11 +49,11 @@ class RESTResponse(io.IOBase):
 
     def getheaders(self):
         """Returns a dictionary of the response headers."""
-        return self.urllib3_response.getheaders()
+        return self.urllib3_response.headers
 
     def getheader(self, name, default=None):
         """Returns a given response header."""
-        return self.urllib3_response.getheader(name, default)
+        return self.urllib3_response.headers.get(name, default)
 
 
 class RESTClientObject(object):

lightning_sdk/machine.py

@@ -92,6 +92,7 @@ class Machine:
     accelerator_count: Optional[int] = None
     cost: Optional[float] = None
     interruptible_cost: Optional[float] = None
+    provider: Optional[str] = None
     wait_time: Optional[float] = None
     interruptible_wait_time: Optional[float] = None
     _include_in_cli: bool = True

lightning_sdk/mmt/base.py

@@ -69,7 +69,7 @@ class _BaseMMT(_BaseJob):
         interruptible: bool = False,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
@@ -104,8 +104,10 @@ class _BaseMMT(_BaseJob):
             cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
                 Required if the registry is part of a cloud provider (e.g. ECR).
             entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
-                just runs the provided command in a standard shell.
-                To use the pre-defined entrypoint of the provided image, set this to an empty string.
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
                 Only applicable when submitting docker jobs.
             path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value
                 represents the data-connection name and the path inside that connection.
@@ -173,7 +175,7 @@ class _BaseMMT(_BaseJob):
                     "Other jobs will automatically persist artifacts to the teamspace distributed filesystem."
                 )
 
-            if entrypoint != "sh -c":
+            if entrypoint is not None:
                 raise ValueError("Specifying the entrypoint has no effect for jobs with Studio envs.")
 
         else:
@@ -199,6 +201,20 @@ class _BaseMMT(_BaseJob):
                     f"<CONNECTION_TYPE>:<CONNECTION_NAME>:<PATH_WITHIN_CONNECTION>, got {artifacts_local}"
                 )
 
+            # command specified, so use the default entrypoint of sh -c
+            if command is not None and entrypoint is None:
+                entrypoint = "sh -c"
+
+            # entrypoint specifically set to empty string, so set to None here to fall back to the image entrypoint
+            elif entrypoint == "":  # noqa: SIM114
+                entrypoint = None
+
+            # entrypoint not specified, but also no command specified, so use the image entrypoint
+            elif entrypoint is None:
+                entrypoint = None
+
+            # all other cases, the entrypoint has been specifically set, so use it as is
+
         inst = cls(name=name, teamspace=teamspace, org=org, user=user, _fetch_job=False)
         inst._submit(
             num_machines=num_machines,
@@ -235,7 +251,7 @@ class _BaseMMT(_BaseJob):
         cloud_provider: Optional[Union["CloudProvider", str]] = None,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str] = None,  # deprecated in favor of path_mappings
@@ -260,8 +276,11 @@ class _BaseMMT(_BaseJob):
                 This should be the name of the respective credentials secret created on the Lightning AI platform.
             cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
                 Required if the registry is part of a cloud provider (e.g. ECR).
-            entrypoint: The entrypoint of your docker container. Defaults to sh -c.
-                To use the pre-defined entrypoint of the provided image, set this to an empty string.
+            entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
                 Only applicable when submitting docker jobs.
             path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value
                 represents the data-connection name and the path inside that connection.

lightning_sdk/mmt/mmt.py

@@ -100,7 +100,7 @@ class MMT(_BaseMMT):
         interruptible: bool = False,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,
@@ -134,8 +134,11 @@ class MMT(_BaseMMT):
                 This should be the name of the respective credentials secret created on the Lightning AI platform.
             cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
                 Required if the registry is part of a cloud provider (e.g. ECR).
-            entrypoint: The entrypoint of your docker container. Defaults to sh -c.
-                To use the pre-defined entrypoint of the provided image, set this to an empty string.
+            entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
                 Only applicable when submitting docker jobs.
             path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value
                 represents the data-connection name and the path inside that connection.
@@ -194,7 +197,7 @@ class MMT(_BaseMMT):
         cloud_provider: Optional[Union["CloudProvider", str]] = None,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
@@ -224,8 +227,10 @@ class MMT(_BaseMMT):
             cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
                 Required if the registry is part of a cloud provider (e.g. ECR).
             entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
-                just runs the provided command in a standard shell.
-                To use the pre-defined entrypoint of the provided image, set this to an empty string.
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
                 Only applicable when submitting docker jobs.
             path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value
                 represents the data-connection name and the path inside that connection.

lightning_sdk/mmt/v1.py

@@ -53,7 +53,7 @@ class _MMTV1(_BaseMMT):
         cloud_provider: Optional[Union["CloudProvider", str]] = None,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,
@@ -89,7 +89,10 @@ class _MMTV1(_BaseMMT):
                 Note that the connection needs to be added to the teamspace already in order for it to be found.
                 Only supported for jobs with a docker image compute environment.
             entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
-                just runs the provided command in a standard shell.
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
                 To use the pre-defined entrypoint of the provided image, set this to an empty string.
                 Only applicable when submitting docker jobs.
             path_mappings: The mappings from data connection inside your container (not supported)

lightning_sdk/mmt/v2.py

@@ -53,7 +53,7 @@ class _MMTV2(_BaseMMT):
         cloud_provider: Optional[Union["CloudProvider", str]] = None,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
@@ -83,7 +83,10 @@ class _MMTV2(_BaseMMT):
             cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
                 Required if the registry is part of a cloud provider (e.g. ECR).
             entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
-                just runs the provided command in a standard shell.
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
                 To use the pre-defined entrypoint of the provided image, set this to an empty string.
                 Only applicable when submitting docker jobs.
             path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value