diff --git a/cvat-sdk/cvat_sdk/core/proxies/projects.py b/cvat-sdk/cvat_sdk/core/proxies/projects.py index 63a9fa37cc5a..05646ee89d11 100644 --- a/cvat-sdk/cvat_sdk/core/proxies/projects.py +++ b/cvat-sdk/cvat_sdk/core/proxies/projects.py @@ -54,6 +54,7 @@ def import_dataset( DatasetUploader(self._client).upload_file_and_wait( self.api.create_dataset_endpoint, + self.api.retrieve_dataset_endpoint, filename, format_name, url_params={"id": self.id}, diff --git a/cvat-sdk/cvat_sdk/core/uploading.py b/cvat-sdk/cvat_sdk/core/uploading.py index 721a2038a842..c6f592f79fba 100644 --- a/cvat-sdk/cvat_sdk/core/uploading.py +++ b/cvat-sdk/cvat_sdk/core/uploading.py @@ -325,7 +325,8 @@ def upload_file_and_wait( class DatasetUploader(Uploader): def upload_file_and_wait( self, - endpoint: Endpoint, + upload_endpoint: Endpoint, + retrieve_endpoint: Endpoint, filename: Path, format_name: str, *, @@ -333,12 +334,14 @@ def upload_file_and_wait( pbar: Optional[ProgressReporter] = None, status_check_period: Optional[int] = None, ): - url = self._client.api_map.make_endpoint_url(endpoint.path, kwsub=url_params) + url = self._client.api_map.make_endpoint_url(upload_endpoint.path, kwsub=url_params) params = {"format": format_name, "filename": filename.name} self.upload_file( url, filename, pbar=pbar, query_params=params, meta={"filename": params["filename"]} ) + url = self._client.api_map.make_endpoint_url(retrieve_endpoint.path, kwsub=url_params) + params = {"action": "import_status"} self._wait_for_completion( url, success_status=201,