Fix missing argument

ysimonson · ysimonson · commit 815332c451df · 2020-10-19T20:13:41.000-04:00
diff --git a/indradb/models.py b/indradb/models.py
@@ -363,7 +363,7 @@ def to_message(self):
 class BulkInsertVertex(_BaseModel):
     __slots__ = ["vertex", "props"]
 
-    def __init__(self, vertex):
+    def __init__(self, vertex, props):
         self.vertex = vertex
         self.props = props
 
@@ -381,7 +381,7 @@ def to_message(self):
 class BulkInsertEdge(_BaseModel):
     __slots__ = ["key", "props"]
 
-    def __init__(self, key):
+    def __init__(self, key, props):
         self.key = key
         self.props = props
 
