fix: tricky check

arangoml · Aug 16, 2024 · 2cd3f5e · 2cd3f5e
1 parent 07ce3a7
commit 2cd3f5e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/adbnx_adapter/adapter.py b/adbnx_adapter/adapter.py
@@ -360,7 +360,7 @@ def networkx_to_arangodb(
                 )
 
                 # 2. Insert batch of nodes
-                if i % node_batch_size == 0:
+                if i and i % node_batch_size == 0:
                     self.__insert_adb_docs(
                         spinner_progress, adb_docs, use_async, **adb_import_kwargs
                     )
@@ -401,7 +401,7 @@ def networkx_to_arangodb(
                 )
 
                 # 2. Insert batch of edges
-                if i % edge_batch_size == 0:
+                if i and i % edge_batch_size == 0:
                     self.__insert_adb_docs(
                         spinner_progress, adb_docs, use_async, **adb_import_kwargs
                     )