diff --git a/cognite/extractorutils/__init__.py b/cognite/extractorutils/__init__.py index f9e8b7a1..b4556360 100644 --- a/cognite/extractorutils/__init__.py +++ b/cognite/extractorutils/__init__.py @@ -16,5 +16,5 @@ Cognite extractor utils is a Python package that simplifies the development of new extractors. """ -__version__ = "6.2.0" +__version__ = "6.2.1" from .base import Extractor diff --git a/cognite/extractorutils/uploader/files.py b/cognite/extractorutils/uploader/files.py index ed615bf8..6921cf33 100644 --- a/cognite/extractorutils/uploader/files.py +++ b/cognite/extractorutils/uploader/files.py @@ -64,7 +64,7 @@ class IOFileUploadQueue(AbstractUploadQueue): def __init__( self, cdf_client: CogniteClient, - post_upload_function: Optional[Callable[[List[Event]], None]] = None, + post_upload_function: Optional[Callable[[List[FileMetadata]], None]] = None, max_queue_size: Optional[int] = None, max_upload_interval: Optional[int] = None, trigger_log_level: str = "DEBUG", @@ -128,7 +128,7 @@ def upload(self) -> None: self.files_written.inc(self.upload_queue_size) try: - self._post_upload(self.upload_queue) + self._post_upload([el[0] for el in self.upload_queue]) except Exception as e: self.logger.error("Error in upload callback: %s", str(e)) self.upload_queue.clear() diff --git a/pyproject.toml b/pyproject.toml index 1879dc49..6b34d28b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "cognite-extractor-utils" -version = "6.2.0" +version = "6.2.1" description = "Utilities for easier development of extractors for CDF" authors = ["Mathias Lohne "] license = "Apache-2.0"