optimize collection validation

arangoml · Jul 22, 2022 · 4f510c5 · 4f510c5
1 parent c7e56fc
commit 4f510c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/adbnx_adapter/adapter.py b/adbnx_adapter/adapter.py
@@ -322,7 +322,7 @@ def networkx_to_arangodb(
                 else self.__cntrl._identify_networkx_node(nx_id, nx_node, adb_v_cols)
             )
 
-            if col not in adb_v_cols:
+            if not has_one_vcol and col not in adb_v_cols:
                 msg = f"'{nx_id}' identified as '{col}', which is not in {adb_v_cols}"
                 raise ValueError(msg)
 
@@ -373,7 +373,7 @@ def networkx_to_arangodb(
                 )
             )
 
-            if col not in adb_e_cols:
+            if not has_one_ecol and col not in adb_e_cols:
                 msg = f"{edge_str} identified as '{col}', which is not in {adb_e_cols}"
                 raise ValueError(msg)