Spaces:
Running
Running
Add timeout
Browse files```python
def wait(self, job):
job_result = job
start_wait = time.time()
while job_result['status'] not in ['succeeded', 'failed']:
if int(time.time() - start_wait) > 60:
raise Exception(f"Timeout error: {job_result['status']}")
time.sleep(0.25)
job_result = self.get_job(job['job'])
return job_result
```html
app.py
CHANGED
|
@@ -35,8 +35,11 @@ class Prodia:
|
|
| 35 |
|
| 36 |
def wait(self, job):
|
| 37 |
job_result = job
|
|
|
|
| 38 |
|
| 39 |
while job_result['status'] not in ['succeeded', 'failed']:
|
|
|
|
|
|
|
| 40 |
time.sleep(0.25)
|
| 41 |
job_result = self.get_job(job['job'])
|
| 42 |
|
|
|
|
| 35 |
|
| 36 |
def wait(self, job):
|
| 37 |
job_result = job
|
| 38 |
+
start_wait = time.time()
|
| 39 |
|
| 40 |
while job_result['status'] not in ['succeeded', 'failed']:
|
| 41 |
+
if int(time.time() - start_wait) > 60:
|
| 42 |
+
raise Exception(f"Timeout error: {job_result['status']}")
|
| 43 |
time.sleep(0.25)
|
| 44 |
job_result = self.get_job(job['job'])
|
| 45 |
|