chore: format

lavague-ai · Aug 28, 2024 · 43c2f8d · 43c2f8d
1 parent c66253d
commit 43c2f8d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/lavague-core/lavague/core/navigation.py b/lavague-core/lavague/core/navigation.py
@@ -158,10 +158,7 @@ def get_nodes(self, query: str) -> List[str]:
 
         html = self.driver.get_html()
 
-        with time_profiler(
-            "Retriever Inference",
-            html_size=len(html),
-        ) as profiler:
+        with time_profiler("Retriever Inference", html_size=len(html)) as profiler:
             source_nodes = self.retriever.retrieve(
                 QueryBundle(query_str=query), [html], viewport_only
             )

diff --git a/lavague-core/lavague/core/utilities/profiling.py b/lavague-core/lavague/core/utilities/profiling.py
@@ -28,10 +28,7 @@ def clear_profiling_data():
 
 @contextmanager
 def time_profiler(
-    event_name,
-    prompt_size=None,
-    html_size=None,
-    full_step_profiling=False,
+    event_name, prompt_size=None, html_size=None, full_step_profiling=False
 ):
     """
     A context manager to profile the execution time of code blocks.