Skip to content

Commit

Permalink
chore: remove matching engine for tests (#1147)
Browse files Browse the repository at this point in the history
  • Loading branch information
morgandu authored Apr 6, 2022
1 parent f2e70b1 commit 31b4f86
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions tests/system/aiplatform/e2e_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -136,10 +136,7 @@ def tear_down_resources(self, shared_state: Dict[str, Any]):
# Bring all Endpoints to the front of the list
# Ensures Models are undeployed first before we attempt deletion
shared_state["resources"].sort(
key=lambda r: 1
if isinstance(r, aiplatform.Endpoint)
or isinstance(r, aiplatform.MatchingEngineIndexEndpoint)
else 2
key=lambda r: 1 if isinstance(r, aiplatform.Endpoint) else 2
)

for resource in shared_state["resources"]:
Expand All @@ -149,7 +146,6 @@ def tear_down_resources(self, shared_state: Dict[str, Any]):
(
aiplatform.Endpoint,
aiplatform.Featurestore,
aiplatform.MatchingEngineIndexEndpoint,
),
):
# For endpoint, undeploy model then delete endpoint
Expand Down

0 comments on commit 31b4f86

Please sign in to comment.