Merge pull request #13 from danielballan/fix-close-on-stop

FIX: Actually close on stop
bluesky · Jun 7, 2019 · 2da8318 · 2da8318
2 parents 1ef4f50 + 121cb05
commit 2da8318
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/suitcase/jsonl/__init__.py b/suitcase/jsonl/__init__.py
@@ -186,11 +186,10 @@ def __call__(self, name, doc):
         self._output_file.write(line)
         if self._flush:
             self._output_file.flush()
+        if name == 'stop':
+            self.close()
         return name, doc
 
-    def stop(self, doc):
-        self.close()
-
     def close(self):
         self._manager.close()