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)