Use the correct signature for exit(...).

usc-isi-i2 · Aug 25, 2020 · 4c905d9 · 4c905d9
1 parent 552da16
commit 4c905d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kgtk/cli/import_wikidata.py b/kgtk/cli/import_wikidata.py
@@ -419,7 +419,7 @@ def enter(self):
                     lineterminator=csv_line_terminator)
 
 
-        def exit(self):
+        def exit(self, *args, **kwargs):
             if self.node_f is not None:
                 self.node_f.close()
             if self.edge_f is not None: