kaggle 1.7.4.2__py3-none-any.whl → 1.8.0__py3-none-any.whl

kagglesdk/kernels/types/kernels_enums.py

@@ -1,5 +1,11 @@
 import enum
 
+class KernelExecutionType(enum.Enum):
+  KERNEL_EXECUTION_TYPE_UNSPECIFIED = 0
+  SAVE_AND_RUN_ALL = 1
+  INTER_ACTIVE = 2
+  QUICK_SAVE = 3
+
 class KernelsListSortType(enum.Enum):
   HOTNESS = 0
   COMMENT_COUNT = 1

kagglesdk/kernels/types/search_kernels.py

@@ -0,0 +1,6 @@
+import enum
+
+class SearchKernelsOrderBy(enum.Enum):
+  SEARCH_KERNELS_ORDER_BY_UNSPECIFIED = 0
+  SEARCH_KERNELS_ORDER_BY_TOTAL_COMMENTS = 1
+

kagglesdk/licenses/types/licenses_types.py

@@ -0,0 +1,182 @@
+from datetime import datetime
+import enum
+from kagglesdk.kaggle_object import *
+from typing import Optional
+
+class UserLicenseAgreementStatus(enum.Enum):
+  r"""
+  This enum tracks the state of a user's agreement with respect to any
+  license that requires it. For Llama 2, users will consent and after some
+  time, Meta will indicate to us that they're approved.
+  """
+  USER_LICENSE_AGREEMENT_STATUS_UNSPECIFIED = 0
+  """Default, unspecified."""
+  USER_LICENSE_AGREEMENT_STATUS_PENDING = 1
+  r"""
+  Users in this status are waiting for review from the 3rd-party. Users in
+  any specified status implicitly have consented to this review. The presence
+  of a ConsentTime value in the UserLicenseAgreements table explicitly
+  denotes if and when the user consented.
+  """
+  USER_LICENSE_AGREEMENT_STATUS_ACCEPTED = 2
+  """3rd-party has approved the user's license agreement."""
+  USER_LICENSE_AGREEMENT_STATUS_REJECTED = 3
+  """3rd-party has rejected the user's license agreement."""
+  USER_LICENSE_AGREEMENT_STATUS_EXPIRED = 4
+  r"""
+  Too much time has passed from when the user submitted the consent form and
+  the license owner reviewed it
+  """
+
+class License(KaggleObject):
+  r"""
+  This proto is a subset of LicenseOption representing the currently selected
+  License. It excludes any metadata needed to organize a selection of options
+  (like display_sequence, etc.)
+
+  Attributes:
+    id (int)
+    name (str)
+    url (str)
+    agreement_required (bool)
+      Indicates whether this license requires agreement to its terms (e.g. Llama
+      2, other custom license, etc.)
+    agreement_status (UserLicenseAgreementStatus)
+      Represents the status of the current user's agreement regarding this
+      specific license. Only applicable when requires_agreement is true
+    consent_time (datetime)
+      If applicable, this is the time when the current user consented to the
+      license agreement
+    current_revision_number (int)
+      If applicable, the current license revision number we want users to be
+      accepting.
+  """
+
+  def __init__(self):
+    self._id = 0
+    self._name = ""
+    self._url = None
+    self._agreement_required = None
+    self._agreement_status = None
+    self._consent_time = None
+    self._current_revision_number = None
+    self._freeze()
+
+  @property
+  def id(self) -> int:
+    return self._id
+
+  @id.setter
+  def id(self, id: int):
+    if id is None:
+      del self.id
+      return
+    if not isinstance(id, int):
+      raise TypeError('id must be of type int')
+    self._id = id
+
+  @property
+  def name(self) -> str:
+    return self._name
+
+  @name.setter
+  def name(self, name: str):
+    if name is None:
+      del self.name
+      return
+    if not isinstance(name, str):
+      raise TypeError('name must be of type str')
+    self._name = name
+
+  @property
+  def url(self) -> str:
+    return self._url or ""
+
+  @url.setter
+  def url(self, url: Optional[str]):
+    if url is None:
+      del self.url
+      return
+    if not isinstance(url, str):
+      raise TypeError('url must be of type str')
+    self._url = url
+
+  @property
+  def agreement_required(self) -> bool:
+    r"""
+    Indicates whether this license requires agreement to its terms (e.g. Llama
+    2, other custom license, etc.)
+    """
+    return self._agreement_required or False
+
+  @agreement_required.setter
+  def agreement_required(self, agreement_required: Optional[bool]):
+    if agreement_required is None:
+      del self.agreement_required
+      return
+    if not isinstance(agreement_required, bool):
+      raise TypeError('agreement_required must be of type bool')
+    self._agreement_required = agreement_required
+
+  @property
+  def agreement_status(self) -> 'UserLicenseAgreementStatus':
+    r"""
+    Represents the status of the current user's agreement regarding this
+    specific license. Only applicable when requires_agreement is true
+    """
+    return self._agreement_status or UserLicenseAgreementStatus.USER_LICENSE_AGREEMENT_STATUS_UNSPECIFIED
+
+  @agreement_status.setter
+  def agreement_status(self, agreement_status: Optional['UserLicenseAgreementStatus']):
+    if agreement_status is None:
+      del self.agreement_status
+      return
+    if not isinstance(agreement_status, UserLicenseAgreementStatus):
+      raise TypeError('agreement_status must be of type UserLicenseAgreementStatus')
+    self._agreement_status = agreement_status
+
+  @property
+  def consent_time(self) -> datetime:
+    r"""
+    If applicable, this is the time when the current user consented to the
+    license agreement
+    """
+    return self._consent_time or None
+
+  @consent_time.setter
+  def consent_time(self, consent_time: Optional[datetime]):
+    if consent_time is None:
+      del self.consent_time
+      return
+    if not isinstance(consent_time, datetime):
+      raise TypeError('consent_time must be of type datetime')
+    self._consent_time = consent_time
+
+  @property
+  def current_revision_number(self) -> int:
+    r"""
+    If applicable, the current license revision number we want users to be
+    accepting.
+    """
+    return self._current_revision_number or 0
+
+  @current_revision_number.setter
+  def current_revision_number(self, current_revision_number: Optional[int]):
+    if current_revision_number is None:
+      del self.current_revision_number
+      return
+    if not isinstance(current_revision_number, int):
+      raise TypeError('current_revision_number must be of type int')
+    self._current_revision_number = current_revision_number
+
+
+License._fields = [
+  FieldMetadata("id", "id", "_id", int, 0, PredefinedSerializer()),
+  FieldMetadata("name", "name", "_name", str, "", PredefinedSerializer()),
+  FieldMetadata("url", "url", "_url", str, None, PredefinedSerializer(), optional=True),
+  FieldMetadata("agreementRequired", "agreement_required", "_agreement_required", bool, None, PredefinedSerializer(), optional=True),
+  FieldMetadata("agreementStatus", "agreement_status", "_agreement_status", UserLicenseAgreementStatus, None, EnumSerializer(), optional=True),
+  FieldMetadata("consentTime", "consent_time", "_consent_time", datetime, None, DateTimeSerializer(), optional=True),
+  FieldMetadata("currentRevisionNumber", "current_revision_number", "_current_revision_number", int, None, PredefinedSerializer(), optional=True),
+]
+

kagglesdk/models/services/model_api_service.py

@@ -1,6 +1,6 @@
 from kagglesdk.common.types.http_redirect import HttpRedirect
 from kagglesdk.kaggle_http_client import KaggleHttpClient
-from kagglesdk.models.types.model_api_service import ApiCreateModelInstanceRequest, ApiCreateModelInstanceVersionRequest, ApiCreateModelRequest, ApiCreateModelResponse, ApiDeleteModelInstanceRequest, ApiDeleteModelInstanceVersionRequest, ApiDeleteModelRequest, ApiDeleteModelResponse, ApiDownloadModelInstanceVersionRequest, ApiGetModelInstanceRequest, ApiGetModelRequest, ApiListModelGatingUserConsentsRequest, ApiListModelGatingUserConsentsResponse, ApiListModelInstanceVersionFilesRequest, ApiListModelInstanceVersionFilesResponse, ApiListModelsRequest, ApiListModelsResponse, ApiModel, ApiModelInstance, ApiReviewGatingUserConsentRequest, ApiUpdateModelInstanceRequest, ApiUpdateModelRequest, ApiUpdateModelResponse, ApiUploadModelFileRequest, ApiUploadModelFileResponse, CreateModelSigningTokenRequest, CreateModelSigningTokenResponse, KeysRequest, KeysResponse, WellKnowEndpointRequest, WellKnowEndpointResponse
+from kagglesdk.models.types.model_api_service import ApiCreateModelInstanceRequest, ApiCreateModelInstanceVersionRequest, ApiCreateModelRequest, ApiCreateModelResponse, ApiDeleteModelInstanceRequest, ApiDeleteModelInstanceVersionRequest, ApiDeleteModelRequest, ApiDeleteModelResponse, ApiDownloadModelInstanceVersionRequest, ApiGetModelInstanceRequest, ApiGetModelRequest, ApiListModelGatingUserConsentsRequest, ApiListModelGatingUserConsentsResponse, ApiListModelInstancesRequest, ApiListModelInstancesResponse, ApiListModelInstanceVersionFilesRequest, ApiListModelInstanceVersionFilesResponse, ApiListModelInstanceVersionsRequest, ApiListModelInstanceVersionsResponse, ApiListModelsRequest, ApiListModelsResponse, ApiModel, ApiModelInstance, ApiReviewModelGatingUserConsentRequest, ApiUpdateModelInstanceRequest, ApiUpdateModelRequest, ApiUpdateModelResponse, ApiUploadModelFileRequest, ApiUploadModelFileResponse, CreateModelSigningTokenRequest, CreateModelSigningTokenResponse, KeysRequest, KeysResponse, WellKnowEndpointRequest, WellKnowEndpointResponse
 
 class ModelApiClient(object):
 
@@ -17,7 +17,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiCreateModelRequest()
 
-    return self._client.call("models.ModelApiService", "ApiCreateModel", request, ApiCreateModelResponse)
+    return self._client.call("models.ModelApiService", "CreateModel", request, ApiCreateModelResponse)
 
   def create_model_instance(self, request: ApiCreateModelInstanceRequest = None) -> ApiCreateModelResponse:
     r"""
@@ -29,7 +29,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiCreateModelInstanceRequest()
 
-    return self._client.call("models.ModelApiService", "ApiCreateModelInstance", request, ApiCreateModelResponse)
+    return self._client.call("models.ModelApiService", "CreateModelInstance", request, ApiCreateModelResponse)
 
   def create_model_instance_version(self, request: ApiCreateModelInstanceVersionRequest = None) -> ApiCreateModelResponse:
     r"""
@@ -41,7 +41,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiCreateModelInstanceVersionRequest()
 
-    return self._client.call("models.ModelApiService", "ApiCreateModelInstanceVersion", request, ApiCreateModelResponse)
+    return self._client.call("models.ModelApiService", "CreateModelInstanceVersion", request, ApiCreateModelResponse)
 
   def delete_model(self, request: ApiDeleteModelRequest = None) -> ApiDeleteModelResponse:
     r"""
@@ -53,7 +53,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiDeleteModelRequest()
 
-    return self._client.call("models.ModelApiService", "ApiDeleteModel", request, ApiDeleteModelResponse)
+    return self._client.call("models.ModelApiService", "DeleteModel", request, ApiDeleteModelResponse)
 
   def delete_model_instance(self, request: ApiDeleteModelInstanceRequest = None) -> ApiDeleteModelResponse:
     r"""
@@ -65,7 +65,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiDeleteModelInstanceRequest()
 
-    return self._client.call("models.ModelApiService", "ApiDeleteModelInstance", request, ApiDeleteModelResponse)
+    return self._client.call("models.ModelApiService", "DeleteModelInstance", request, ApiDeleteModelResponse)
 
   def delete_model_instance_version(self, request: ApiDeleteModelInstanceVersionRequest = None) -> ApiDeleteModelResponse:
     r"""
@@ -77,7 +77,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiDeleteModelInstanceVersionRequest()
 
-    return self._client.call("models.ModelApiService", "ApiDeleteModelInstanceVersion", request, ApiDeleteModelResponse)
+    return self._client.call("models.ModelApiService", "DeleteModelInstanceVersion", request, ApiDeleteModelResponse)
 
   def get_model(self, request: ApiGetModelRequest = None) -> ApiModel:
     r"""
@@ -89,7 +89,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiGetModelRequest()
 
-    return self._client.call("models.ModelApiService", "ApiGetModel", request, ApiModel)
+    return self._client.call("models.ModelApiService", "GetModel", request, ApiModel)
 
   def get_model_instance(self, request: ApiGetModelInstanceRequest = None) -> ApiModelInstance:
     r"""
@@ -101,7 +101,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiGetModelInstanceRequest()
 
-    return self._client.call("models.ModelApiService", "ApiGetModelInstance", request, ApiModelInstance)
+    return self._client.call("models.ModelApiService", "GetModelInstance", request, ApiModelInstance)
 
   def download_model_instance_version(self, request: ApiDownloadModelInstanceVersionRequest = None) -> HttpRedirect:
     r"""
@@ -113,7 +113,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiDownloadModelInstanceVersionRequest()
 
-    return self._client.call("models.ModelApiService", "ApiDownloadModelInstanceVersion", request, HttpRedirect)
+    return self._client.call("models.ModelApiService", "DownloadModelInstanceVersion", request, HttpRedirect)
 
   def list_models(self, request: ApiListModelsRequest = None) -> ApiListModelsResponse:
     r"""
@@ -125,7 +125,31 @@ class ModelApiClient(object):
     if request is None:
       request = ApiListModelsRequest()
 
-    return self._client.call("models.ModelApiService", "ApiListModels", request, ApiListModelsResponse)
+    return self._client.call("models.ModelApiService", "ListModels", request, ApiListModelsResponse)
+
+  def list_model_instances(self, request: ApiListModelInstancesRequest = None) -> ApiListModelInstancesResponse:
+    r"""
+    Args:
+      request (ApiListModelInstancesRequest):
+        The request object; initialized to empty instance if not specified.
+    """
+
+    if request is None:
+      request = ApiListModelInstancesRequest()
+
+    return self._client.call("models.ModelApiService", "ListModelInstances", request, ApiListModelInstancesResponse)
+
+  def list_model_instance_versions(self, request: ApiListModelInstanceVersionsRequest = None) -> ApiListModelInstanceVersionsResponse:
+    r"""
+    Args:
+      request (ApiListModelInstanceVersionsRequest):
+        The request object; initialized to empty instance if not specified.
+    """
+
+    if request is None:
+      request = ApiListModelInstanceVersionsRequest()
+
+    return self._client.call("models.ModelApiService", "ListModelInstanceVersions", request, ApiListModelInstanceVersionsResponse)
 
   def list_model_instance_version_files(self, request: ApiListModelInstanceVersionFilesRequest = None) -> ApiListModelInstanceVersionFilesResponse:
     r"""
@@ -137,7 +161,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiListModelInstanceVersionFilesRequest()
 
-    return self._client.call("models.ModelApiService", "ApiListModelInstanceVersionFiles", request, ApiListModelInstanceVersionFilesResponse)
+    return self._client.call("models.ModelApiService", "ListModelInstanceVersionFiles", request, ApiListModelInstanceVersionFilesResponse)
 
   def update_model(self, request: ApiUpdateModelRequest = None) -> ApiUpdateModelResponse:
     r"""
@@ -149,7 +173,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiUpdateModelRequest()
 
-    return self._client.call("models.ModelApiService", "ApiUpdateModel", request, ApiUpdateModelResponse)
+    return self._client.call("models.ModelApiService", "UpdateModel", request, ApiUpdateModelResponse)
 
   def update_model_instance(self, request: ApiUpdateModelInstanceRequest = None) -> ApiUpdateModelResponse:
     r"""
@@ -161,7 +185,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiUpdateModelInstanceRequest()
 
-    return self._client.call("models.ModelApiService", "ApiUpdateModelInstance", request, ApiUpdateModelResponse)
+    return self._client.call("models.ModelApiService", "UpdateModelInstance", request, ApiUpdateModelResponse)
 
   def upload_model_file(self, request: ApiUploadModelFileRequest = None) -> ApiUploadModelFileResponse:
     r"""
@@ -175,7 +199,7 @@ class ModelApiClient(object):
     if request is None:
       request = ApiUploadModelFileRequest()
 
-    return self._client.call("models.ModelApiService", "ApiUploadModelFile", request, ApiUploadModelFileResponse)
+    return self._client.call("models.ModelApiService", "UploadModelFile", request, ApiUploadModelFileResponse)
 
   def create_model_signing_token(self, request: CreateModelSigningTokenRequest = None) -> CreateModelSigningTokenResponse:
     r"""
@@ -238,18 +262,19 @@ class ModelApiClient(object):
     if request is None:
       request = ApiListModelGatingUserConsentsRequest()
 
-    return self._client.call("models.ModelApiService", "ApiListModelGatingUserConsents", request, ApiListModelGatingUserConsentsResponse)
+    return self._client.call("models.ModelApiService", "ListModelGatingUserConsents", request, ApiListModelGatingUserConsentsResponse)
 
-  def review_gating_user_consent(self, request: ApiReviewGatingUserConsentRequest = None):
+  def review_model_gating_user_consent(self, request: ApiReviewModelGatingUserConsentRequest = None):
     r"""
-    Review the user consents for an agreement.
+    Review the user consents for a gated model, under the model's current
+    active agreement.
 
     Args:
-      request (ApiReviewGatingUserConsentRequest):
+      request (ApiReviewModelGatingUserConsentRequest):
         The request object; initialized to empty instance if not specified.
     """
 
     if request is None:
-      request = ApiReviewGatingUserConsentRequest()
+      request = ApiReviewModelGatingUserConsentRequest()
 
-    self._client.call("models.ModelApiService", "ApiReviewGatingUserConsent", request, None)
+    self._client.call("models.ModelApiService", "ReviewModelGatingUserConsent", request, None)