fix: Adds sleep hold

e4e_data_management/cli.py

@@ -11,6 +11,8 @@
 from pathlib import Path
 from typing import Callable, List, Optional, TypeVar
 
+from wakepy import keep
+
 from e4e_data_management import __version__
 from e4e_data_management.core import DataManager
 from e4e_data_management.metadata import Metadata
@@ -280,7 +282,8 @@ def main(self):
             arg_fn = args.func
             arg_dict.pop('func')
 
-            arg_fn(**arg_dict)
+            with keep.running():
+                arg_fn(**arg_dict)
         except Exception as exc:
             self._log.exception('Exception during main execution')
             raise exc

pyproject.toml

@@ -13,6 +13,7 @@ readme = "README.md"
 python = "^3.11"
 schema = "^0.7.7"
 appdirs = "^1.4.4"
+wakepy = "^0.10.1"
 
 
 [tool.poetry.group.dev.dependencies]