diff --git a/src/tidy_tools/frame/__init__.py b/src/tidy_tools/frame/__init__.py index e11dbbc..a17996b 100644 --- a/src/tidy_tools/frame/__init__.py +++ b/src/tidy_tools/frame/__init__.py @@ -1,4 +1,5 @@ +from tidy_tools.frame.context import tidycontext as tidycontext from tidy_tools.frame.dataframe import TidyDataFrame as TidyDataFrame -from tidy_tools.frame.workflow import TidyWorkFlow as TidyWorkFlow +from tidy_tools.frame.workflow import tidyworkflow as tidyworkflow -__all__ = ["TidyDataFrame", "TidyWorkFlow"] +__all__ = ["TidyDataFrame", "tidyworkflow", "tidycontext"] diff --git a/src/tidy_tools/frame/context.py b/src/tidy_tools/frame/context.py index a61a459..361a049 100644 --- a/src/tidy_tools/frame/context.py +++ b/src/tidy_tools/frame/context.py @@ -8,8 +8,8 @@ from pyspark.sql import types as T -logger.level("ENTER", color="") -logger.level("EXIT", color="") +logger.level("ENTER", no=98, color="") +logger.level("EXIT", no=99, color="") @define @@ -22,7 +22,7 @@ class TidySnapshot: @contextmanager -def tidy_context(): +def tidycontext(): """Define context manager for handling tidy operations.""" context = {"operation_log": [], "snapshots": []} try: