51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
from __future__ import print_function
|
|
|
|
import logging
|
|
import time
|
|
|
|
import config as config
|
|
from utils.api import Api
|
|
|
|
logger = logging.getLogger(__name__)
|
|
logger.setLevel(logging.INFO)
|
|
|
|
|
|
class TestCleanInferences:
|
|
def setup_class(self):
|
|
self.api = Api(config)
|
|
self.api.feat_oas_schema()
|
|
|
|
@classmethod
|
|
def teardown_class(self):
|
|
pass
|
|
|
|
def test_1_clean_all_inferences(self):
|
|
headers = {
|
|
"x-api-key": config.api_key,
|
|
"username": config.username
|
|
}
|
|
|
|
while True:
|
|
|
|
resp = self.api.list_inferences(headers=headers, params={"limit": 20})
|
|
inferences = resp.json()['data']['inferences']
|
|
if len(inferences) == 0:
|
|
break
|
|
|
|
ids = []
|
|
i = 0
|
|
for inference in inferences:
|
|
i = i + 1
|
|
inference_id = inference['InferenceJobId']
|
|
ids.append(inference_id)
|
|
logger.info(f"delete execute {i} {inference_id}")
|
|
|
|
data = {
|
|
"inference_id_list": ids,
|
|
}
|
|
resp = self.api.delete_inferences(headers=headers, data=data)
|
|
if resp.status_code == 400:
|
|
logger.info(resp.json()['message'])
|
|
time.sleep(5)
|
|
continue
|