forked from Azure/azureml-examples
-
Notifications
You must be signed in to change notification settings - Fork 0
/
cleanup.py
29 lines (25 loc) · 868 Bytes
/
cleanup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# imports
import argparse
from azureml.core import Workspace
# setup argparse
parser = argparse.ArgumentParser()
parser.add_argument("--config", type=str, default="")
args = parser.parse_args()
# get workspace
ws = Workspace.from_config(args.config)
# delete all webservices
for webservice in ws.webservices:
ws.webservices[webservice].delete()
# delete some compute targets
for compute_target in ws.compute_targets:
if ws.compute_targets[compute_target].get_status() in ["Failed", "Canceled", None]:
try:
ws.compute_targets[compute_target].delete()
except:
pass
elif (
"dask-ct" in compute_target
and ws.compute_targets[compute_target].get_status() in ["Succeeded"]
and len(ws.compute_targets[compute_target].list_nodes()) == 0
):
ws.compute_targets[compute_target].delete()