Skip to content

Commit 530763f

Browse files
authored
Update sdk in src (#700)
Generated code.
1 parent be063c2 commit 530763f

File tree

2 files changed

+38
-0
lines changed

2 files changed

+38
-0
lines changed

src/kagglesdk/kernels/types/kernels_api_service.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1216,6 +1216,9 @@ class ApiSaveKernelRequest(KaggleObject):
12161216
`{username}/{model-slug}/{framework}/{variation-slug}`
12171217
Or versioned:
12181218
`{username}/{model-slug}/{framework}/{variation-slug}/{version-number}`
1219+
session_timeout_seconds (int)
1220+
If specified, terminate the kernel session after this many seconds of
1221+
runtime, which must be lower than the global maximum.
12191222
"""
12201223

12211224
def __init__(self):
@@ -1235,6 +1238,7 @@ def __init__(self):
12351238
self._enable_internet = None
12361239
self._docker_image_pinning_type = None
12371240
self._model_data_sources = []
1241+
self._session_timeout_seconds = None
12381242
self._freeze()
12391243

12401244
@property
@@ -1496,6 +1500,23 @@ def model_data_sources(self, model_data_sources: Optional[List[str]]):
14961500
raise TypeError('model_data_sources must contain only items of type str')
14971501
self._model_data_sources = model_data_sources
14981502

1503+
@property
1504+
def session_timeout_seconds(self) -> int:
1505+
r"""
1506+
If specified, terminate the kernel session after this many seconds of
1507+
runtime, which must be lower than the global maximum.
1508+
"""
1509+
return self._session_timeout_seconds or 0
1510+
1511+
@session_timeout_seconds.setter
1512+
def session_timeout_seconds(self, session_timeout_seconds: int):
1513+
if session_timeout_seconds is None:
1514+
del self.session_timeout_seconds
1515+
return
1516+
if not isinstance(session_timeout_seconds, int):
1517+
raise TypeError('session_timeout_seconds must be of type int')
1518+
self._session_timeout_seconds = session_timeout_seconds
1519+
14991520
def endpoint(self):
15001521
path = '/api/v1/kernels/push'
15011522
return path.format_map(self.to_field_map(self))
@@ -1902,6 +1923,7 @@ def creation_date(self, creation_date: str):
19021923
FieldMetadata("enableInternet", "enable_internet", "_enable_internet", bool, None, PredefinedSerializer(), optional=True),
19031924
FieldMetadata("dockerImagePinningType", "docker_image_pinning_type", "_docker_image_pinning_type", str, None, PredefinedSerializer(), optional=True),
19041925
FieldMetadata("modelDataSources", "model_data_sources", "_model_data_sources", str, [], ListSerializer(PredefinedSerializer())),
1926+
FieldMetadata("sessionTimeoutSeconds", "session_timeout_seconds", "_session_timeout_seconds", int, None, PredefinedSerializer(), optional=True),
19051927
]
19061928

19071929
ApiSaveKernelResponse._fields = [

src/kagglesdk/models/types/model_types.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,7 @@ class Owner(KaggleObject):
141141
profile_url (str)
142142
slug (str)
143143
user_tier (UserAchievementTier)
144+
allow_model_gating (bool)
144145
"""
145146

146147
def __init__(self):
@@ -151,6 +152,7 @@ def __init__(self):
151152
self._profile_url = None
152153
self._slug = ""
153154
self._user_tier = UserAchievementTier.NOVICE
155+
self._allow_model_gating = None
154156
self._freeze()
155157

156158
@property
@@ -244,6 +246,19 @@ def user_tier(self, user_tier: 'UserAchievementTier'):
244246
raise TypeError('user_tier must be of type UserAchievementTier')
245247
self._user_tier = user_tier
246248

249+
@property
250+
def allow_model_gating(self) -> bool:
251+
return self._allow_model_gating or False
252+
253+
@allow_model_gating.setter
254+
def allow_model_gating(self, allow_model_gating: bool):
255+
if allow_model_gating is None:
256+
del self.allow_model_gating
257+
return
258+
if not isinstance(allow_model_gating, bool):
259+
raise TypeError('allow_model_gating must be of type bool')
260+
self._allow_model_gating = allow_model_gating
261+
247262

248263
BaseModelInstanceInformation._fields = [
249264
FieldMetadata("id", "id", "_id", int, 0, PredefinedSerializer()),
@@ -266,5 +281,6 @@ def user_tier(self, user_tier: 'UserAchievementTier'):
266281
FieldMetadata("profileUrl", "profile_url", "_profile_url", str, None, PredefinedSerializer(), optional=True),
267282
FieldMetadata("slug", "slug", "_slug", str, "", PredefinedSerializer()),
268283
FieldMetadata("userTier", "user_tier", "_user_tier", UserAchievementTier, UserAchievementTier.NOVICE, EnumSerializer()),
284+
FieldMetadata("allowModelGating", "allow_model_gating", "_allow_model_gating", bool, None, PredefinedSerializer(), optional=True),
269285
]
270286

0 commit comments

Comments
 (0)