diff --git a/scrapegraphai/integrations/burr_bridge.py b/scrapegraphai/integrations/burr_bridge.py index d1fe566f..e5eb3c6a 100644 --- a/scrapegraphai/integrations/burr_bridge.py +++ b/scrapegraphai/integrations/burr_bridge.py @@ -11,12 +11,13 @@ import inspect try: import burr + from burr import tracking + from burr.core import Application, ApplicationBuilder, State, Action, default, ApplicationContext + from burr.lifecycle import PostRunStepHook, PreRunStepHook except ImportError: raise ImportError("burr package is not installed. Please install it with 'pip install scrapegraphai[burr]'") -from burr import tracking -from burr.core import Application, ApplicationBuilder, State, Action, default, ApplicationContext -from burr.lifecycle import PostRunStepHook, PreRunStepHook + class PrintLnHook(PostRunStepHook, PreRunStepHook):