trying to setup context manager

duguyue100 · Mar 16, 2024 · 319c05c · 319c05c
1 parent 9f010e1
commit 319c05c
Show file tree

Hide file tree

Showing 3 changed files with 80 additions and 291 deletions.
diff --git a/pyaer/__init__.py b/pyaer/__init__.py
@@ -6,11 +6,16 @@
 LOG_LEVEL = log.DEBUG
 
 try:
-    from pyaer import libcaer_wrap as libcaer  # noqa
+    from pyaer import libcaer_wrap as libcaer
 except ImportError:
     raise ImportError(
         "libcaer might not be in the LD_LIBRARY_PATH "
         "or your numpy might not be the required version. "
         "Try to load _libcaer_wrap.so from the package "
         "directory, this will provide more information."
     )
+
+from pyaer.event_camera import EventCamera  # noqa # noreorder
+
+
+__all__ = ["libcaer", "EventCamera"]