Merge pull request #16 from WarpCloud/fix_schema_version

fix version mismatch for higher version
WarpCloud · Dec 17, 2024 · edb05dd · edb05dd
2 parents 88d8d41 + 5fbd844
commit edb05dd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pystellardb/graph_types.py b/pystellardb/graph_types.py
@@ -241,7 +241,7 @@ def parseEdgeFromDict(schema, m):
         if 'startKey' not in m:
             raise ValueError("Could not find start node entity key in JSON")
 
-        if schema.getVersion() == 18:
+        if schema.getVersion() >= 18:
             startUid = prop_dict['__srcuid']
             startLabelIdx = Vertex.parseLabelIdxFromRKV18(m['startKey'])
         else:
@@ -262,7 +262,7 @@ def parseEdgeFromDict(schema, m):
         if 'endKey' not in m:
             raise ValueError("Could not find end node entity key in JSON")
 
-        if schema.getVersion() == 18:
+        if schema.getVersion() >= 18:
             endUid = prop_dict['__dstuid']
             endLabelIdx = Vertex.parseLabelIdxFromRKV18(m['endKey'])
         else: