fix: expose some missing functions of the common and worker modules

aneoconsulting · Nov 22, 2024 · 21492d4 · 21492d4
1 parent 3e97b32
commit 21492d4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/packages/python/src/armonik/common/__init__.py b/packages/python/src/armonik/common/__init__.py
@@ -1,3 +1,4 @@
+from .channel import create_channel
 from .enumwrapper import (
     Direction,
     EventTypes,
@@ -36,6 +37,7 @@
 from .filter import Filter
 
 __all__ = [
+    "create_channel",
     "datetime_to_timestamp",
     "timestamp_to_datetime",
     "duration_to_timedelta",

diff --git a/packages/python/src/armonik/worker/__init__.py b/packages/python/src/armonik/worker/__init__.py
@@ -1,9 +1,10 @@
-from .worker import ArmoniKWorker
+from .worker import ArmoniKWorker, armonik_worker
 from .taskhandler import TaskHandler
 from .seqlogger import ClefLogger
 
 __all__ = [
     "ArmoniKWorker",
     "TaskHandler",
     "ClefLogger",
+    "armonik_worker",
 ]