From 3b677e5e978fb95f00db5a83ad1b3c1f4af3a1be Mon Sep 17 00:00:00 2001 From: brian Date: Mon, 6 Jan 2025 11:26:23 +0000 Subject: [PATCH] fixed bug in context manager definition --- shared/datastore/src/datastore_interface.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/shared/datastore/src/datastore_interface.py b/shared/datastore/src/datastore_interface.py index 97aa365..7cd5445 100644 --- a/shared/datastore/src/datastore_interface.py +++ b/shared/datastore/src/datastore_interface.py @@ -1,5 +1,7 @@ """Definition of datastore interface.""" +from __future__ import annotations + from abc import ABC, abstractmethod from collections import OrderedDict @@ -25,8 +27,9 @@ class DatastoreInterface(ABC): @abstractmethod def __enter__( self, - ) -> None: + ) -> DatastoreInterface: self.connect() + return self @abstractmethod def __exit__(