meg-huggingface commited on
Commit
5c33832
·
1 Parent(s): 3d16b0d

Handling more exceptions

Browse files
Files changed (1) hide show
  1. src/backend/inference_endpoint.py +2 -3
src/backend/inference_endpoint.py CHANGED
@@ -37,14 +37,13 @@ def create_endpoint(endpoint_name, repository, framework='pytorch',
37
  endpoint.update(repository=repository,
38
  framework=framework, task=task,
39
  accelerator=accelerator,
40
- vendor=vendor, region=region,
41
- type=type,
42
  instance_size=instance_size,
43
  instance_type=instance_type)
44
  except requests.exceptions.HTTPError as e:
45
- # Not enough compute, or wrong compute
46
  logger.debug("Hit error:")
47
  logger.debug(e)
 
48
  endpoint = update_endpoint_exception(endpoint)
49
  except Exception as e:
50
  logger.debug("Hit unaccounted-for error")
 
37
  endpoint.update(repository=repository,
38
  framework=framework, task=task,
39
  accelerator=accelerator,
 
 
40
  instance_size=instance_size,
41
  instance_type=instance_type)
42
  except requests.exceptions.HTTPError as e:
43
+ # Not enough compute, wrong compute, or quota exceeded
44
  logger.debug("Hit error:")
45
  logger.debug(e)
46
+ logger.debug("Attempting a different compute.")
47
  endpoint = update_endpoint_exception(endpoint)
48
  except Exception as e:
49
  logger.debug("Hit unaccounted-for error")