minor

Signed-off-by: Sarad Mohanan <[email protected]>
datafold · Nov 29, 2023 · 933a393 · 933a393
1 parent fe49f40
commit 933a393
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/data_diff/databases/base.py b/data_diff/databases/base.py
@@ -918,6 +918,9 @@ class Database(abc.ABC):
     is_closed: bool = False
     _dialect: BaseDialect = None
 
+    def __enter__(self):
+        return self
+
     def __exit__(self, exc_type, exc_value, traceback):
         self.close()