Merge pull request #191 from amosproj/fix/page-number

fix(page-number): Add 1 to `page_number`
amosproj · Jul 9, 2024 · 16b8906 · 16b8906
2 parents 00ea4d2 + 2c2df83
commit 16b8906
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Project/backend/codebase/graph_creator/services/netx_graphdb.py b/Project/backend/codebase/graph_creator/services/netx_graphdb.py
@@ -36,6 +36,8 @@ def create_graph_from_df(self, data: pd.DataFrame, chunks: dict) -> nx.Graph:
 
             chunk_id = edge["chunk_id"]
             page_number = chunk_to_page[int(chunk_id)]
+            if isinstance(page_number, int):
+                page_number += 1
 
             # Add nodes with page attribute
             if edge["node_1"] not in graph: