We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 861306a commit fa835a3Copy full SHA for fa835a3
nucleus/__init__.py
@@ -1209,8 +1209,13 @@ def create_object_index(
1209
model_run_id: id of the model run for generating embeddings on. Mutually exclusive with gt_only
1210
gt_only: Whether we are generating embeddings on the ground truth objects in a dataset. Mutually exclusive with model_run_id
1211
"""
1212
+ payload: Dict[str, Union[str, bool]] = {}
1213
+ if model_run_id:
1214
+ payload["model_run_id"] = model_run_id
1215
+ elif gt_only:
1216
+ payload["ingest_gt_only"] = True
1217
return self.make_request(
- {model_run_id: model_run_id or None, gt_only: gt_only or None},
1218
+ payload,
1219
f"indexing/{dataset_id}/internal/object",
1220
requests_command=requests.post,
1221
)
0 commit comments