Skip to content

Commit 8a81716

Browse files
authored
Checkin worker for extractors (#464)
1 parent 2302bc4 commit 8a81716

File tree

5 files changed

+864
-5
lines changed

5 files changed

+864
-5
lines changed

cognite/extractorutils/unstable/core/base.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -69,14 +69,15 @@ def my_task_function(self, task_context: TaskContext) -> None:
6969
LogFileHandlerConfig,
7070
)
7171
from cognite.extractorutils.unstable.core._dto import (
72-
Error as DtoError,
73-
)
74-
from cognite.extractorutils.unstable.core._dto import (
72+
CogniteModel,
7573
ExtractorInfo,
7674
StartupRequest,
7775
TaskType,
7876
TaskUpdate,
7977
)
78+
from cognite.extractorutils.unstable.core._dto import (
79+
Error as DtoError,
80+
)
8081
from cognite.extractorutils.unstable.core._dto import (
8182
Task as DtoTask,
8283
)
@@ -88,7 +89,12 @@ def my_task_function(self, task_context: TaskContext) -> None:
8889
from cognite.extractorutils.unstable.scheduling import TaskScheduler
8990
from cognite.extractorutils.util import now
9091

91-
__all__ = ["ConfigRevision", "ConfigType", "Extractor"]
92+
__all__ = [
93+
"CogniteModel",
94+
"ConfigRevision",
95+
"ConfigType",
96+
"Extractor",
97+
]
9298

9399

94100
_T = TypeVar("_T", bound=ExtractorConfig)

0 commit comments

Comments
 (0)