From db34d7a5e4c0a606548ff1bebc56fe3df93b5b06 Mon Sep 17 00:00:00 2001
From: lucas-nelson-uiuc <lucas.nelson.uiuc@gmail.com>
Date: Sat, 30 Nov 2024 13:09:35 -0600
Subject: [PATCH] build: update references

---
 src/tidy_tools/frame/__init__.py | 5 +++--
 src/tidy_tools/frame/context.py  | 6 +++---
 2 files changed, 6 insertions(+), 5 deletions(-)

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="<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: