fix: add pkg __all__

loiccoyle · Sep 22, 2023 · d007276 · d007276
1 parent 6bfcbb4
commit d007276
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/phomo/__init__.py b/phomo/__init__.py
@@ -3,6 +3,9 @@
 from .master import Master
 from .mosaic import Mosaic
 from .pool import Pool
+from .metrics import METRICS
+
+__all__ = ["Master", "Mosaic", "Pool", "METRICS"]
 
 logger = logging.getLogger(__name__)
 logger.addHandler(logging.NullHandler())