diff --git a/datamint/api/endpoints/inference_api.py b/datamint/api/endpoints/inference_api.py index 9487932..f776fe9 100644 --- a/datamint/api/endpoints/inference_api.py +++ b/datamint/api/endpoints/inference_api.py @@ -138,7 +138,7 @@ def submit( model_version: int | None = None, model_alias: str | None = None, resource_id: str | None = None, - # resource_ids: list[str] | None = None, + resource_ids: list[str] | None = None, file_path: str | None = None, file_paths: list[str] | None = None, save_results: bool = False, @@ -174,8 +174,8 @@ def submit( save_results_options=save_results_options, params=params, ) - # if resource_ids is not None: - # payload["resource_ids"] = resource_ids + if resource_ids is not None: + payload["resource_ids"] = resource_ids if file_paths is not None: payload["file_paths"] = file_paths