def _get_instances(self, **kwargs):
try:
instances, meta_data = self.client.list(get_meta=True, **kwargs)
except sdk_exceptions.GradientSdkError as e:
raise exceptions.ReceivingDataFailedError(e)
return instances, meta_data
def _get_instances(self, kwargs):
try:
instances = self.client.list()
except api_sdk.GradientSdkError as e:
raise exceptions.ReceivingDataFailedError(e)
return instances
def _get_instances(self, **kwargs):
filters = self._get_request_json(kwargs)
try:
instances = self.deployment_client.list(filters)
except api_sdk.GradientSdkError as e:
raise exceptions.ReceivingDataFailedError(e)
return instances
def execute(self, **kwargs):
with halo.Halo(text=self.WAITING_FOR_RESPONSE_MESSAGE, spinner="dots"):
try:
instances = self.client.artifacts_list(**kwargs)
except sdk_exceptions.GradientSdkError as e:
raise exceptions.ReceivingDataFailedError(e)
self._log_objects_list(instances, kwargs)
def _get_instances(self, **kwargs):
filters = self._get_request_params(kwargs)
try:
instances = self.client.artifacts_list(filters)
except api_sdk.GradientSdkError as e:
raise exceptions.ReceivingDataFailedError(e)
return instances
def _get_instance(self, id_):
"""
:rtype: api_sdk.Hyperparameter
"""
try:
instance = self.client.get(id_)
except api_sdk.GradientSdkError as e:
raise exceptions.ReceivingDataFailedError(e)
return instance
def _get_instance(self, id_):
"""
:rtype: api_sdk.SingleNodeExperiment|api_sdk.MultiNodeExperiment
"""
try:
instance = self.client.get(id_)
except api_sdk.GradientSdkError as e:
raise exceptions.ReceivingDataFailedError(e)
return instance
def _get_instances(self, **kwargs):
limit = kwargs.get("limit")
offset = kwargs.get("offset")
tags = kwargs.get("tags")
get_meta = True
try:
instances, meta_data = self.client.list(get_meta=get_meta,
limit=limit,
offset=offset,
tags=tags)
except sdk_exceptions.GradientSdkError as e:
raise exceptions.ReceivingDataFailedError(e)
return instances, meta_data
def get_namespace(self):
client = api_sdk.clients.http_client.API(
config.CONFIG_HOST,
api_key=self.api_key,
logger=self.logger,
ps_client_name=CLI_PS_CLIENT_NAME,
)
try:
resp = client.get(url='/teams/namespace')
except sdk_exceptions.GradientSdkError as e:
raise exceptions.ReceivingDataFailedError(e)
if isinstance(resp.json(), str):
return resp.json()
return ''
def execute(self, **kwargs):
with halo.Halo(text=self.WAITING_FOR_RESPONSE_MESSAGE, spinner="dots"):
try:
start_after = None
instances = []
while True:
pagination_response = self.client.artifacts_list(
start_after=start_after, **kwargs)
if pagination_response.data:
instances.extend(pagination_response.data)
start_after = pagination_response.start_after
if start_after is None:
break
except sdk_exceptions.GradientSdkError as e:
raise exceptions.ReceivingDataFailedError(e)
self._log_objects_list(instances, kwargs)
