build: update references

lucas-nelson-uiuc · Nov 30, 2024 · db34d7a · db34d7a
1 parent 9b73b67
commit db34d7a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git 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
@@ -8,8 +8,8 @@
 from pyspark.sql import types as T
 
 
-logger.level("ENTER", color="<green>")
-logger.level("EXIT", color="<red>")
+logger.level("ENTER", no=98, color="<green>")
+logger.level("EXIT", no=99, color="<red>")
 
 
 @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: