cognite-extractor-utils 7.5.6__py3-none-any.whl → 7.5.8__py3-none-any.whl

cognite/extractorutils/unstable/core/runtime.py

@@ -5,8 +5,7 @@ import time
 from argparse import ArgumentParser, Namespace
 from multiprocessing import Process, Queue
 from pathlib import Path
-from typing import Any, Generic, Type, TypeVar
-from uuid import uuid4
+from typing import Any, Generic, TypeVar
 
 from requests.exceptions import ConnectionError
 from typing_extensions import assert_never
@@ -17,7 +16,6 @@ from cognite.extractorutils.unstable.configuration.exceptions import InvalidConf
 from cognite.extractorutils.unstable.configuration.loaders import load_file, load_from_cdf
 from cognite.extractorutils.unstable.configuration.models import ConnectionConfig
 from cognite.extractorutils.unstable.core._dto import Error
-from cognite.extractorutils.util import now
 
 from ._messaging import RuntimeMessage
 from .base import ConfigRevision, ConfigType, Extractor, FullConfig
@@ -30,7 +28,7 @@ ExtractorType = TypeVar("ExtractorType", bound=Extractor)
 class Runtime(Generic[ExtractorType]):
     def __init__(
         self,
-        extractor: Type[ExtractorType],
+        extractor: type[ExtractorType],
     ) -> None:
         self._extractor_class = extractor
         self._cancellation_token = CancellationToken()
@@ -127,15 +125,13 @@ class Runtime(Generic[ExtractorType]):
         self,
         args: Namespace,
         connection_config: ConnectionConfig,
-    ) -> tuple[ConfigType, ConfigRevision, ConfigRevision]:
+    ) -> tuple[ConfigType, ConfigRevision]:
         current_config_revision: ConfigRevision
-        newest_config_revision: ConfigRevision
 
         if args.local_override:
             self.logger.info("Loading local application config")
 
             current_config_revision = "local"
-            newest_config_revision = "local"
             try:
                 application_config = load_file(args.local_override[0], self._extractor_class.CONFIG_TYPE)
             except InvalidConfigError as e:
@@ -153,50 +149,12 @@ class Runtime(Generic[ExtractorType]):
 
             errors: list[Error] = []
 
-            revision: int | None = None
             try:
-                while True:
-                    try:
-                        application_config, current_config_revision = load_from_cdf(
-                            client,
-                            connection_config.integration,
-                            self._extractor_class.CONFIG_TYPE,
-                            revision=revision,
-                        )
-                        break
-
-                    except InvalidConfigError as e:
-                        if e.attempted_revision is None:
-                            # Should never happen, attempted_revision is set in every handler in load_from_cdf, but it's
-                            # needed for type checks to pass
-                            raise e
-
-                        self.logger.error(f"Revision {e.attempted_revision} is invalid: {e.message}")
-
-                        t = now()
-                        errors.append(
-                            Error(
-                                external_id=str(uuid4()),
-                                level="error",
-                                description=f"Revision {e.attempted_revision} is invalid",
-                                details=e.message,
-                                start_time=t,
-                                end_time=t,
-                                task=None,
-                            )
-                        )
-
-                        if revision is None:
-                            revision = e.attempted_revision - 1
-                            newest_config_revision = e.attempted_revision
-                        else:
-                            revision -= 1
-
-                        if revision > 0:
-                            self.logger.info(f"Falling back to revision {revision}")
-                        else:
-                            self.logger.critical("No more revisions to fall back to")
-                            raise e
+                application_config, current_config_revision = load_from_cdf(
+                    client,
+                    connection_config.integration,
+                    self._extractor_class.CONFIG_TYPE,
+                )
 
             finally:
                 if errors:
@@ -209,7 +167,7 @@ class Runtime(Generic[ExtractorType]):
                         headers={"cdf-version": "alpha"},
                     )
 
-        return application_config, current_config_revision, newest_config_revision
+        return application_config, current_config_revision
 
     def _verify_connection_config(self, connection_config: ConnectionConfig) -> bool:
         client = connection_config.get_cognite_client(
@@ -281,9 +239,7 @@ class Runtime(Generic[ExtractorType]):
         application_config: Any
         while not self._cancellation_token.is_cancelled:
             try:
-                application_config, current_config_revision, newest_config_revision = self._get_application_config(
-                    args, connection_config
-                )
+                application_config, current_config_revision = self._get_application_config(args, connection_config)
 
             except InvalidConfigError:
                 self.logger.critical("Could not get a valid application config file. Shutting down")
@@ -295,7 +251,6 @@ class Runtime(Generic[ExtractorType]):
                     connection_config=connection_config,
                     application_config=application_config,
                     current_config_revision=current_config_revision,
-                    newest_config_revision=newest_config_revision,
                 )
             )
             process.join()

cognite/extractorutils/unstable/core/tasks.py

@@ -1,6 +1,6 @@
 from abc import ABC
+from collections.abc import Callable
 from dataclasses import dataclass
-from typing import Callable
 
 from cognite.extractorutils.unstable.configuration.models import ScheduleConfig
 

cognite/extractorutils/unstable/scheduling/_scheduler.py

@@ -1,8 +1,8 @@
+from collections.abc import Callable
 from dataclasses import dataclass
 from logging import getLogger
 from threading import RLock, Thread
 from time import time
-from typing import Callable
 
 import arrow
 from humps import pascalize

cognite/extractorutils/uploader/_base.py

@@ -15,8 +15,9 @@
 import logging
 import threading
 from abc import ABC, abstractmethod
+from collections.abc import Callable
 from dataclasses import dataclass
-from typing import Any, Callable
+from typing import Any
 
 from arrow import Arrow
 

cognite/extractorutils/uploader/assets.py

@@ -12,8 +12,9 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
+from collections.abc import Callable
 from types import TracebackType
-from typing import Any, Callable, Type
+from typing import Any
 
 from cognite.client import CogniteClient
 from cognite.client.data_classes.assets import Asset
@@ -147,7 +148,7 @@ class AssetUploadQueue(AbstractUploadQueue):
 
     def __exit__(
         self,
-        exc_type: Type[BaseException] | None,
+        exc_type: type[BaseException] | None,
         exc_val: BaseException | None,
         exc_tb: TracebackType | None,
     ) -> None:

cognite/extractorutils/uploader/data_modeling.py

@@ -1,5 +1,6 @@
+from collections.abc import Callable
 from types import TracebackType
-from typing import Any, Callable, Type
+from typing import Any
 
 from cognite.client import CogniteClient
 from cognite.client.data_classes.data_modeling import EdgeApply, NodeApply
@@ -100,7 +101,7 @@ class InstanceUploadQueue(AbstractUploadQueue):
 
     def __exit__(
         self,
-        exc_type: Type[BaseException] | None,
+        exc_type: type[BaseException] | None,
         exc_val: BaseException | None,
         exc_tb: TracebackType | None,
     ) -> None:

cognite/extractorutils/uploader/events.py

@@ -12,8 +12,8 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
+from collections.abc import Callable
 from types import TracebackType
-from typing import Callable, Type
 
 from cognite.client import CogniteClient
 from cognite.client.data_classes import Event
@@ -151,7 +151,7 @@ class EventUploadQueue(AbstractUploadQueue):
         return self
 
     def __exit__(
-        self, exc_type: Type[BaseException] | None, exc_val: BaseException | None, exc_tb: TracebackType | None
+        self, exc_type: type[BaseException] | None, exc_val: BaseException | None, exc_tb: TracebackType | None
     ) -> None:
         """
         Wraps around stop method, for use as context manager

cognite/extractorutils/uploader/files.py

@@ -13,6 +13,7 @@
 #  limitations under the License.
 
 import threading
+from collections.abc import Callable, Iterator
 from concurrent.futures import Future, ThreadPoolExecutor
 from io import BytesIO, RawIOBase
 from math import ceil
@@ -21,12 +22,6 @@ from types import TracebackType
 from typing import (
     Any,
     BinaryIO,
-    Callable,
-    Iterator,
-    List,
-    Optional,
-    Type,
-    Union,
 )
 from urllib.parse import ParseResult, urlparse
 
@@ -67,7 +62,7 @@ _MAX_FILE_CHUNK_SIZE = 4 * 1024 * 1024 * 1000
 _CDF_ALPHA_VERSION_HEADER = {"cdf-version": "alpha"}
 
 
-FileMetadataOrCogniteExtractorFile = Union[FileMetadata, CogniteExtractorFileApply]
+FileMetadataOrCogniteExtractorFile = FileMetadata | CogniteExtractorFileApply
 
 
 class ChunkedStream(RawIOBase, BinaryIO):
@@ -111,7 +106,7 @@ class ChunkedStream(RawIOBase, BinaryIO):
 
     def __exit__(
         self,
-        exc_type: Type[BaseException] | None,
+        exc_type: type[BaseException] | None,
         exc_val: BaseException | None,
         exc_tb: TracebackType | None,
     ) -> None:
@@ -146,7 +141,7 @@ class ChunkedStream(RawIOBase, BinaryIO):
         if size > 0:
             self._pos += size
             return self._inner.read(size)
-        return bytes()
+        return b""
 
     def next_chunk(self) -> bool:
         """
@@ -209,8 +204,8 @@ class IOFileUploadQueue(AbstractUploadQueue):
         trigger_log_level: str = "DEBUG",
         thread_name: str | None = None,
         overwrite_existing: bool = False,
-        cancellation_token: Optional[CancellationToken] = None,
-        max_parallelism: Optional[int] = None,
+        cancellation_token: CancellationToken | None = None,
+        max_parallelism: int | None = None,
         failure_logging_path: None | str = None,
         ssl_verify: bool | str = True,
     ):
@@ -231,8 +226,8 @@ class IOFileUploadQueue(AbstractUploadQueue):
         self.failure_logging_path = failure_logging_path or None
         self.initialize_failure_logging()
 
-        self.upload_queue: List[Future] = []
-        self.errors: List[Exception] = []
+        self.upload_queue: list[Future] = []
+        self.errors: list[Exception] = []
 
         self.overwrite_existing = overwrite_existing
 
@@ -429,7 +424,7 @@ class IOFileUploadQueue(AbstractUploadQueue):
         self,
         file_meta: FileMetadataOrCogniteExtractorFile,
         read_file: Callable[[], BinaryIO],
-        extra_retries: tuple[Type[Exception], ...] | dict[Type[Exception], Callable[[Any], bool]] | None = None,
+        extra_retries: tuple[type[Exception], ...] | dict[type[Exception], Callable[[Any], bool]] | None = None,
     ) -> None:
         """
         Add file to upload queue. The file will start uploading immedeately. If the size of the queue is larger than
@@ -584,9 +579,9 @@ class IOFileUploadQueue(AbstractUploadQueue):
 
     def __exit__(
         self,
-        exc_type: Optional[Type[BaseException]],
-        exc_val: Optional[BaseException],
-        exc_tb: Optional[TracebackType],
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: TracebackType | None,
     ) -> None:
         """
         Wraps around stop method, for use as context manager
@@ -649,7 +644,7 @@ class FileUploadQueue(IOFileUploadQueue):
     def add_to_upload_queue(
         self,
         file_meta: FileMetadataOrCogniteExtractorFile,
-        file_name: Union[str, PathLike],
+        file_name: str | PathLike,
     ) -> None:
         """
         Add file to upload queue. The queue will be uploaded if the queue size is larger than the threshold

cognite/extractorutils/uploader/raw.py

@@ -12,8 +12,9 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
+from collections.abc import Callable
 from types import TracebackType
-from typing import Any, Callable, Type
+from typing import Any
 
 import arrow
 from arrow import Arrow
@@ -162,7 +163,7 @@ class RawUploadQueue(AbstractUploadQueue):
         return self
 
     def __exit__(
-        self, exc_type: Type[BaseException] | None, exc_val: BaseException | None, exc_tb: TracebackType | None
+        self, exc_type: type[BaseException] | None, exc_val: BaseException | None, exc_tb: TracebackType | None
     ) -> None:
         """
         Wraps around stop method, for use as context manager

cognite/extractorutils/uploader/time_series.py

@@ -13,9 +13,10 @@
 #  limitations under the License.
 
 import math
+from collections.abc import Callable
 from datetime import datetime
 from types import TracebackType
-from typing import Any, Callable, Type
+from typing import Any
 
 from cognite.client import CogniteClient
 from cognite.client.data_classes import (
@@ -317,7 +318,7 @@ class TimeSeriesUploadQueue(AbstractUploadQueue):
         return self
 
     def __exit__(
-        self, exc_type: Type[BaseException] | None, exc_val: BaseException | None, exc_tb: TracebackType | None
+        self, exc_type: type[BaseException] | None, exc_val: BaseException | None, exc_tb: TracebackType | None
     ) -> None:
         """
         Wraps around stop method, for use as context manager
@@ -474,11 +475,11 @@ class SequenceUploadQueue(AbstractUploadQueue):
         if isinstance(rows, SequenceRows):
             # Already in the desired format
             pass
-        elif isinstance(rows, (dict, list)):
+        elif isinstance(rows, dict | list):
             rows_raw: list[dict[str, Any]]
             if isinstance(rows, dict):
                 rows_raw = [{"rowNumber": row_number, "values": values} for row_number, values in rows.items()]
-            elif isinstance(rows, list) and rows and isinstance(rows[0], (tuple, list)):
+            elif isinstance(rows, list) and rows and isinstance(rows[0], tuple | list):
                 rows_raw = [{"rowNumber": row_number, "values": values} for row_number, values in rows]
             else:
                 rows_raw = rows  # type: ignore[assignment]
@@ -491,7 +492,7 @@ class SequenceUploadQueue(AbstractUploadQueue):
                 }
             )
         else:
-            raise TypeError("Unsupported type for sequence rows: {}".format(type(rows)))
+            raise TypeError(f"Unsupported type for sequence rows: {type(rows)}")
 
         with self.lock:
             seq = self.upload_queue.get(either_id)
@@ -520,7 +521,7 @@ class SequenceUploadQueue(AbstractUploadQueue):
             backoff=RETRY_BACKOFF_FACTOR,
         )
         def _upload_single(either_id: EitherId, upload_this: SequenceData) -> SequenceData:
-            self.logger.debug("Writing {} rows to sequence {}".format(len(upload_this.values), either_id))
+            self.logger.debug(f"Writing {len(upload_this.values)} rows to sequence {either_id}")
 
             try:
                 self.cdf_client.sequences.data.insert(
@@ -595,7 +596,7 @@ class SequenceUploadQueue(AbstractUploadQueue):
             )
 
         except CogniteDuplicatedError:
-            self.logger.info("Sequnce already exist: {}".format(either_id))
+            self.logger.info(f"Sequnce already exist: {either_id}")
             seq = self.cdf_client.sequences.retrieve(  # type: ignore [assignment]
                 id=either_id.internal_id,
                 external_id=either_id.external_id,
@@ -656,7 +657,7 @@ class SequenceUploadQueue(AbstractUploadQueue):
         return self
 
     def __exit__(
-        self, exc_type: Type[BaseException] | None, exc_val: BaseException | None, exc_tb: TracebackType | None
+        self, exc_type: type[BaseException] | None, exc_val: BaseException | None, exc_tb: TracebackType | None
     ) -> None:
         """
         Wraps around stop method, for use as context manager

cognite/extractorutils/uploader/upload_failure_handler.py

@@ -1,5 +1,5 @@
+from collections.abc import Iterator
 from datetime import datetime
-from typing import Iterator, List
 
 import jsonlines
 
@@ -9,7 +9,7 @@ class FileErrorMapping:
         self.file_name = file_name
         self.error_reason = error_reason
 
-    def __iter__(self) -> Iterator[List[str]]:
+    def __iter__(self) -> Iterator[list[str]]:
         return iter([[self.file_name, self.error_reason]])
 
 

cognite/extractorutils/uploader_extractor.py

@@ -16,9 +16,10 @@
 A module containing a slightly more advanced base extractor class, sorting a generic output into upload queues.
 """
 
+from collections.abc import Callable, Iterable
 from dataclasses import dataclass
 from types import TracebackType
-from typing import Any, Callable, Iterable, Type, TypeVar
+from typing import Any, TypeVar
 
 from more_itertools import peekable
 
@@ -80,7 +81,7 @@ class UploaderExtractor(Extractor[UploaderExtractorConfigClass]):
         description: str,
         version: str | None = None,
         run_handle: RunHandle | None = None,
-        config_class: Type[UploaderExtractorConfigClass],
+        config_class: type[UploaderExtractorConfigClass],
         metrics: BaseMetrics | None = None,
         use_default_state_store: bool = True,
         cancellation_token: CancellationToken | None = None,
@@ -90,7 +91,7 @@ class UploaderExtractor(Extractor[UploaderExtractorConfigClass]):
         handle_interrupts: bool = True,
         middleware: list[Callable[[dict], dict]] | None = None,
     ):
-        super(UploaderExtractor, self).__init__(
+        super().__init__(
             name=name,
             description=description,
             version=version,
@@ -144,7 +145,7 @@ class UploaderExtractor(Extractor[UploaderExtractorConfigClass]):
         return item
 
     def __enter__(self) -> "UploaderExtractor":
-        super(UploaderExtractor, self).__enter__()
+        super().__enter__()
 
         queue_config = self.config.queues if self.config.queues else QueueConfigClass()
         self.event_queue = EventUploadQueue(
@@ -170,9 +171,9 @@ class UploaderExtractor(Extractor[UploaderExtractorConfigClass]):
         return self
 
     def __exit__(
-        self, exc_type: Type[BaseException] | None, exc_val: BaseException | None, exc_tb: TracebackType | None
+        self, exc_type: type[BaseException] | None, exc_val: BaseException | None, exc_tb: TracebackType | None
     ) -> bool:
         self.event_queue.__exit__(exc_type, exc_val, exc_tb)
         self.raw_queue.__exit__(exc_type, exc_val, exc_tb)
         self.time_series_queue.__exit__(exc_type, exc_val, exc_tb)
-        return super(UploaderExtractor, self).__exit__(exc_type, exc_val, exc_tb)
+        return super().__exit__(exc_type, exc_val, exc_tb)

cognite/extractorutils/uploader_types.py

@@ -1,4 +1,5 @@
-from typing import Iterable, TypeAlias
+from collections.abc import Iterable
+from typing import TypeAlias
 
 from cognite.client.data_classes import Event as _Event
 from cognite.client.data_classes import Row as _Row

cognite/extractorutils/util.py

@@ -20,12 +20,13 @@ extractors.
 import io
 import logging
 import random
+from collections.abc import Callable, Generator, Iterable
 from datetime import datetime, timezone
 from functools import partial, wraps
 from io import RawIOBase
 from threading import Thread
 from time import time
-from typing import Any, Callable, Generator, Iterable, Type, TypeVar
+from typing import Any, TypeVar
 
 from decorator import decorator
 
@@ -157,7 +158,7 @@ class EitherId:
         Returns:
             A string rep of the EitherId
         """
-        return "{}: {}".format(self.type(), self.content())
+        return f"{self.type()}: {self.content()}"
 
     def __repr__(self) -> str:
         """
@@ -313,7 +314,7 @@ _T2 = TypeVar("_T2")
 def _retry_internal(
     f: Callable[..., _T2],
     cancellation_token: CancellationToken,
-    exceptions: tuple[Type[Exception], ...] | dict[Type[Exception], Callable[[Exception], bool]],
+    exceptions: tuple[type[Exception], ...] | dict[type[Exception], Callable[[Exception], bool]],
     tries: int,
     delay: float,
     max_delay: float | None,
@@ -367,7 +368,7 @@ def _retry_internal(
 
 def retry(
     cancellation_token: CancellationToken | None = None,
-    exceptions: tuple[Type[Exception], ...] | dict[Type[Exception], Callable[[Any], bool]] = (Exception,),
+    exceptions: tuple[type[Exception], ...] | dict[type[Exception], Callable[[Any], bool]] = (Exception,),
     tries: int = 10,
     delay: float = 1,
     max_delay: float | None = 60,
@@ -415,7 +416,7 @@ def retry(
 
 def requests_exceptions(
     status_codes: list[int] | None = None,
-) -> dict[Type[Exception], Callable[[Any], bool]]:
+) -> dict[type[Exception], Callable[[Any], bool]]:
     """
     Retry exceptions from using the ``requests`` library. This will retry all connection and HTTP errors matching
     the given status codes.
@@ -449,7 +450,7 @@ def requests_exceptions(
 
 def httpx_exceptions(
     status_codes: list[int] | None = None,
-) -> dict[Type[Exception], Callable[[Any], bool]]:
+) -> dict[type[Exception], Callable[[Any], bool]]:
     """
     Retry exceptions from using the ``httpx`` library. This will retry all connection and HTTP errors matching
     the given status codes.
@@ -483,7 +484,7 @@ def httpx_exceptions(
 
 def cognite_exceptions(
     status_codes: list[int] | None = None,
-) -> dict[Type[Exception], Callable[[Any], bool]]:
+) -> dict[type[Exception], Callable[[Any], bool]]:
     """
     Retry exceptions from using the Cognite SDK. This will retry all connection and HTTP errors matching
     the given status codes.
@@ -499,7 +500,7 @@ def cognite_exceptions(
     status_codes = status_codes or [408, 425, 429, 500, 502, 503, 504]
 
     def handle_cognite_errors(exception: CogniteException) -> bool:
-        if isinstance(exception, (CogniteAPIError, CogniteFileUploadError)):
+        if isinstance(exception, CogniteAPIError | CogniteFileUploadError):
             return exception.code in status_codes
         return True
 

{cognite_extractor_utils-7.5.6.dist-info → cognite_extractor_utils-7.5.8.dist-info}/METADATA

@@ -1,39 +1,34 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: cognite-extractor-utils
-Version: 7.5.6
+Version: 7.5.8
 Summary: Utilities for easier development of extractors for CDF
-Home-page: https://github.com/cognitedata/python-extractor-utils
+Project-URL: repository, https://github.com/cognitedata/python-extractor-utils
+Author-email: Mathias Lohne <mathias.lohne@cognite.com>
 License: Apache-2.0
-Author: Mathias Lohne
-Author-email: mathias.lohne@cognite.com
-Requires-Python: >=3.9.0,<4.0.0
-Classifier: License :: OSI Approved :: Apache Software License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
+License-File: LICENSE
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Programming Language :: Python
+Requires-Python: >=3.10
+Requires-Dist: arrow>=1.0.0
+Requires-Dist: azure-identity>=1.14.0
+Requires-Dist: azure-keyvault-secrets>=4.7.0
+Requires-Dist: cognite-sdk>=7.59.0
+Requires-Dist: croniter>=6.0.0
+Requires-Dist: dacite>=1.6.0
+Requires-Dist: decorator>=5.1.1
+Requires-Dist: httpx<1,>=0.27.0
+Requires-Dist: jsonlines>=4.0.0
+Requires-Dist: more-itertools>=10.0.0
+Requires-Dist: orjson>=3.10.3
+Requires-Dist: prometheus-client<=1.0.0,>=0.7.0
+Requires-Dist: psutil>=6.0.0
+Requires-Dist: pydantic>=2.8.2
+Requires-Dist: pyhumps>=3.8.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: pyyaml<7,>=5.3.0
+Requires-Dist: typing-extensions<5,>=3.7.4
 Provides-Extra: experimental
-Requires-Dist: arrow (>=1.0.0,<2.0.0)
-Requires-Dist: azure-identity (>=1.14.0,<2.0.0)
-Requires-Dist: azure-keyvault-secrets (>=4.7.0,<5.0.0)
-Requires-Dist: cognite-sdk (>=7.59.0,<8.0.0)
-Requires-Dist: croniter (>=6.0.0,<7.0.0)
-Requires-Dist: dacite (>=1.6.0,<2.0.0)
-Requires-Dist: decorator (>=5.1.1,<6.0.0)
-Requires-Dist: httpx (>=0.27.0,<0.28.0)
-Requires-Dist: jsonlines (>=4.0.0,<5.0.0)
-Requires-Dist: more-itertools (>=10.0.0,<11.0.0)
-Requires-Dist: orjson (>=3.10.3,<4.0.0)
-Requires-Dist: prometheus-client (>0.7.0,<=1.0.0)
-Requires-Dist: psutil (>=6.0.0,<7.0.0)
-Requires-Dist: pydantic (>=2.8.2,<3.0.0)
-Requires-Dist: pyhumps (>=3.8.0,<4.0.0)
-Requires-Dist: python-dotenv (>=1.0.0,<2.0.0)
-Requires-Dist: pyyaml (>=5.3.0,<7)
-Requires-Dist: typing-extensions (>=3.7.4,<5)
-Project-URL: Repository, https://github.com/cognitedata/python-extractor-utils
+Requires-Dist: cognite-sdk-experimental; extra == 'experimental'
 Description-Content-Type: text/markdown
 
 <a href="https://cognite.com/">
@@ -96,10 +91,10 @@ as a code of conduct.
 
 ### Development environment
 
-We use [poetry](https://python-poetry.org) to manage dependencies and to administrate virtual environments. To develop
+We use [uv](https://docs.astral.sh/uv/) to manage dependencies and to administrate virtual environments. To develop
 `extractor-utils`, follow the following steps to set up your local environment:
 
- 1. [Install poetry](https://python-poetry.org/docs/#installation) if you haven't already.
+ 1. [Install uv](https://docs.astral.sh/uv/getting-started/installation/) if you haven't already.
 
  2. Clone repository:
     ```
@@ -111,7 +106,7 @@ We use [poetry](https://python-poetry.org) to manage dependencies and to adminis
     ```
  4. Create virtual environment and install dependencies:
     ```
-    $ poetry install
+    $ uv sync
     ```
 
 
@@ -127,4 +122,3 @@ $ poetry run pre-commit install
 Each public method, class and module should have docstrings. Docstrings are written in the [Google
 style](https://google.github.io/styleguide/pyguide.html#38-comments-and-docstrings). Please include unit and/or
 integration tests for submitted code, and remember to update the [changelog](./CHANGELOG.md).
-