diff --git a/mockfirestore/client.py b/mockfirestore/client.py index e39d82e..4bd7c67 100644 --- a/mockfirestore/client.py +++ b/mockfirestore/client.py @@ -60,6 +60,8 @@ def transaction(self, **kwargs) -> Transaction: return Transaction(self, **kwargs) def batch(self) -> Transaction: - return Transaction(self) + batch = Transaction(self) + batch._begin() + return batch