edbeeching commited on
Commit
5d89dcd
·
1 Parent(s): 60d17e2

fix num samples in validation

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -53,7 +53,7 @@ class GenerationRequest:
53
  private: bool = False
54
  num_retries: int = 0
55
 
56
- def validate_request(request: GenerationRequest):
57
  # checks that the request is valid
58
  # - input dataset exists and can be accessed with the provided token
59
  try:
@@ -99,6 +99,8 @@ def validate_request(request: GenerationRequest):
99
  # check valid email address TODO: use py3-validate-email https://stackoverflow.com/questions/8022530/how-to-check-for-valid-email-address
100
  if "@" not in request.email or "." not in request.email.split("@")[-1]:
101
  raise Exception("Invalid email address")
 
 
102
 
103
 
104
  def add_request_to_db(request: GenerationRequest):
@@ -248,7 +250,7 @@ def create_gradio_interface():
248
  )
249
 
250
  # check the input dataset exists and can be accessed with the provided token
251
- validate_request(request)
252
  add_request_to_db(request)
253
 
254
  return "Request submitted successfully!"
 
53
  private: bool = False
54
  num_retries: int = 0
55
 
56
+ def validate_request(request: GenerationRequest) -> GenerationRequest:
57
  # checks that the request is valid
58
  # - input dataset exists and can be accessed with the provided token
59
  try:
 
99
  # check valid email address TODO: use py3-validate-email https://stackoverflow.com/questions/8022530/how-to-check-for-valid-email-address
100
  if "@" not in request.email or "." not in request.email.split("@")[-1]:
101
  raise Exception("Invalid email address")
102
+
103
+ return request
104
 
105
 
106
  def add_request_to_db(request: GenerationRequest):
 
250
  )
251
 
252
  # check the input dataset exists and can be accessed with the provided token
253
+ request = validate_request(request)
254
  add_request_to_db(request)
255
 
256
  return "Request submitted successfully!"