Round time in debug log

Author: stellasia

examples/build_graph/simple_kg_builder_from_pdf.py

@@ -22,6 +22,9 @@
 AUTH = ("neo4j", "password")
 DATABASE = "neo4j"
 
+import logging
+
+logging.basicConfig(level=logging.DEBUG)
 
 root_dir = Path(__file__).parents[1]
 file_path = root_dir / "data" / "Harry Potter and the Chamber of Secrets Summary.pdf"

src/neo4j_graphrag/experimental/pipeline/pipeline.py

@@ -104,7 +104,7 @@ async def run(
         res = await self.execute(context, inputs)
         end_time = default_timer()
         logger.debug(
-            f"TASK FINISHED {self.name} in {end_time - start_time} res={prettify(res)}"
+            f"TASK FINISHED {self.name} in {round(end_time - start_time)}s res={prettify(res)}"
         )
         return res
 
@@ -567,7 +567,7 @@ async def run(self, data: dict[str, Any]) -> PipelineResult:
             )
             raise e
         end_time = default_timer()
-        logger.debug(f"PIPELINE FINISHED {run_id} in {end_time - start_time}s")
+        logger.debug(f"PIPELINE FINISHED {run_id} in {round(end_time - start_time)}s")
         return res
 
     async def _run(self, run_id: str, data: dict[str, Any]) -> PipelineResult: