Revert "Close datastore after task_finished has been invoked (Netflix…

…#757)" (Netflix#783) This reverts commit 8a54a42.
AAAWallet · Oct 25, 2021 · eee086a · eee086a
1 parent 2c75494
commit eee086a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/metaflow/task.py b/metaflow/task.py
@@ -496,6 +496,10 @@ def run_step(self,
             output.save_metadata({'task_end': {}})
             output.persist(self.flow)
 
+            # this writes a success marker indicating that the
+            # "transaction" is done
+            output.done()
+
             # final decorator hook: The task results are now
             # queryable through the client API / datastore
             for deco in decorators:
@@ -506,10 +510,6 @@ def run_step(self,
                                    retry_count,
                                    max_user_code_retries)
 
-            # this writes a success marker indicating that the
-            # "transaction" is done
-            output.done()
-
             # terminate side cars
             logger.terminate()
             self.metadata.stop_heartbeat()