From aae26b67d9777c2bf307a0e3810d08bde988cfd5 Mon Sep 17 00:00:00 2001 From: Andrey Date: Tue, 7 Mar 2023 18:12:22 +0200 Subject: [PATCH] Fix import. --- db/moonstreamdb/db.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/db/moonstreamdb/db.py b/db/moonstreamdb/db.py index 4eeae18e..c32c70b1 100644 --- a/db/moonstreamdb/db.py +++ b/db/moonstreamdb/db.py @@ -3,9 +3,10 @@ Moonstream database connection. """ from contextlib import contextmanager import os +from typing import Generator from sqlalchemy import create_engine -from sqlalchemy.orm import sessionmaker, Session, Generator +from sqlalchemy.orm import sessionmaker, Session MOONSTREAM_DB_URI = os.environ.get("MOONSTREAM_DB_URI") if MOONSTREAM_DB_URI is None: