Skip to content

Commit d7bfeda

Browse files
Keep endpoint configs after tests
1 parent 773457e commit d7bfeda

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/test_clean_core.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def test_clean_inference(request):
6060
predicted_value = predictor.predict(data=[1])
6161
assert predicted_value == [43]
6262

63-
predictor.delete_endpoint()
63+
predictor.delete_endpoint(delete_endpoint_config=False)
6464

6565

6666
# noinspection DuplicatedCode
@@ -137,7 +137,8 @@ def test_clean_inference_mms(request, instance_type):
137137
assert predicted_value == [20043]
138138

139139
finally:
140-
predictor.delete_endpoint()
140+
if predictor:
141+
predictor.delete_endpoint(delete_endpoint_config=False)
141142

142143

143144
# noinspection DuplicatedCode
@@ -225,4 +226,4 @@ def test_clean_inference_mms_without_model(request, instance_type):
225226
assert predicted_value == [20043]
226227

227228
finally:
228-
predictor.delete_endpoint()
229+
predictor.delete_endpoint(delete_endpoint_config=False)

0 commit comments

Comments
 (0)