Skip to content

Commit 330a78c

Browse files
authored
fix: Engine running check in RM (#378)
1 parent 0faba0e commit 330a78c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/firebolt/model/V2/engine.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,8 @@ def check_attached_to_database(func: Callable) -> Callable:
3535

3636
@functools.wraps(func)
3737
def inner(self: Engine, *args: Any, **kwargs: Any) -> Any:
38-
if self.database is None:
38+
# this check is only relevant for account version 1
39+
if self._service.client._account_version != 2 and self.database is None:
3940
raise NoAttachedDatabaseError(method_name=func.__name__)
4041
return func(self, *args, **kwargs)
4142

0 commit comments

Comments
 (0)