Skip to content

Commit bb83716

Browse files
committed
Clean remove_iam_integration()
1 parent b177143 commit bb83716

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

libs/labelbox/src/labelbox/schema/dataset.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -999,18 +999,15 @@ def remove_iam_integration(self) -> None:
999999
mutation DetachSignerPyApi($id: ID!) {
10001000
clearSignerForDataset(where: { id: $id }) {
10011001
id
1002-
signer {
1003-
id
1004-
}
10051002
}
10061003
}
10071004
"""
10081005

10091006
response = self.client.execute(query, {"id": self.uid})
10101007

1011-
if response:
1012-
return response.get("clearSignerForDataset", {}).get("signer")
1013-
else:
1014-
raise ResourceNotFoundError(IAMIntegration, {"datasetId": self.uid})
1008+
if not response:
1009+
raise ResourceNotFoundError(Dataset, {"id": self.uid})
1010+
1011+
10151012

10161013

0 commit comments

Comments
 (0)