diff --git a/src/sweets/__main__.py b/src/sweets/__main__.py index 3a7c34c..3c99193 100644 --- a/src/sweets/__main__.py +++ b/src/sweets/__main__.py @@ -1,4 +1,5 @@ """Main module to provide command line interface to the workflows.""" + import sys from .cli import main diff --git a/src/sweets/_log.py b/src/sweets/_log.py index 78ed46f..387c00d 100644 --- a/src/sweets/_log.py +++ b/src/sweets/_log.py @@ -15,6 +15,7 @@ # Custom output for this module: logger.success("Something great happened: highlight this success") """ + import logging import time from collections.abc import Callable diff --git a/src/sweets/_show_versions.py b/src/sweets/_show_versions.py index 6b92374..dc2c2e5 100644 --- a/src/sweets/_show_versions.py +++ b/src/sweets/_show_versions.py @@ -4,6 +4,7 @@ which was adapted from `sklearn.utils._show_versions` which was adapted from `pandas.show_versions` """ + from __future__ import annotations import importlib