Skip to content

Commit 602ce16

Browse files
author
Matt Sokoloff
committed
check for valid integration
1 parent 36c911b commit 602ce16

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

labelbox/client.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -526,10 +526,15 @@ def create_dataset(self, **kwargs):
526526
if iam_integration is not None:
527527
if not isinstance(iam_integration, IAMIntegration):
528528
raise TypeError(f"iam integration must be a reference an `IAMIntegration` object. Found {iam_integration}")
529+
530+
if not iam_integration.valid:
531+
raise ValueError("Invalid integration is invalid. Please select another integration or remove default.")
532+
529533
try:
530534
import time
531535
time.sleep(5)
532-
print(iam_integration.uid)
536+
537+
533538
res = self.execute("""mutation setSignerForDatasetPyApi($signerId: ID!, $datasetId: ID!) { setSignerForDataset(data: { signerId: $signerId}, where: {id: $datasetId}){id}} """, {'signerId' : iam_integration.uid, 'datasetId' : dataset.uid})
534539
#result = self.execute("""mutation validateDatasetPyApi($id: ID!){validateDataset(where: {id : $id}){valid checks{name, success}} }""", {'id' : dataset.uid})
535540
# TODO: I am not sure what to do with this check...

0 commit comments

Comments
 (0)