Expose get_daemon_client so it can be imported from aiida.engine (#…

…6008) This is public API that users should use to easily obtain a `DaemonClient` instance to control the daemon. Currently it has to be imported from `aiida.engine.daemon.client` which is unnecessarily buried deep down.
aiidateam · May 11, 2023 · 1a0c1ee · 1a0c1ee
1 parent 2bfc3c6
commit 1a0c1ee
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/aiida/engine/__init__.py b/aiida/engine/__init__.py
@@ -61,6 +61,7 @@
     'assign_',
     'calcfunction',
     'construct_awaitable',
+    'get_daemon_client',
     'get_object_loader',
     'if_',
     'interruptable_task',

diff --git a/aiida/engine/daemon/__init__.py b/aiida/engine/daemon/__init__.py
@@ -18,6 +18,7 @@
 
 __all__ = (
     'DaemonClient',
+    'get_daemon_client',
 )
 
 # yapf: enable
diff --git a/aiida/engine/daemon/client.py b/aiida/engine/daemon/client.py
@@ -41,7 +41,7 @@
 # Recent versions of virtualenv create the environment variable VIRTUAL_ENV
 VIRTUALENV = os.environ.get('VIRTUAL_ENV', None)
 
-__all__ = ('DaemonClient',)
+__all__ = ('DaemonClient', 'get_daemon_client')
 
 
 class ControllerProtocol(enum.Enum):