fix: Update imports and add __version__ attribute

jjjermiah · Feb 3, 2024 · daad13e · daad13e
1 parent 8bcdfa4
commit daad13e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/nbiatoolkit/__init__.py b/src/nbiatoolkit/__init__.py
@@ -8,11 +8,11 @@
 
 
 # import the modules
-from .nbia import NBIAClient
-from .nbia_cli import version, getCollections_cli
+from .nbia import NBIAClient, __version__
+from .nbia_cli import version
 from .auth import OAuth2
 from .logger.logger import setup_logger
 from .utils.nbia_endpoints import NBIA_ENDPOINTS
 
 # define the __all__ variable
-__all__ = ["NBIAClient", "OAuth2", "setup_logger", "NBIA_ENDPOINTS", "version"]
+__all__ = ["NBIAClient", "OAuth2", "setup_logger", "NBIA_ENDPOINTS", "version", "__version__"]