From eb9636cae15889c8027f63f0e9cc9d4797dd0ed3 Mon Sep 17 00:00:00 2001 From: beverage Date: Fri, 19 Jul 2024 15:20:37 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=B9=20Some=20minor=20import=20and=20op?= =?UTF-8?q?tions=20cleanup?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/dbtest/dbtest.py | 2 +- src/dbtest/sentences/create.py | 1 - src/dbtest/sentences/models.py | 2 +- 3 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/dbtest/dbtest.py b/src/dbtest/dbtest.py index 6da6746..68266a4 100755 --- a/src/dbtest/dbtest.py +++ b/src/dbtest/dbtest.py @@ -20,7 +20,7 @@ @click.option('--debug', default=False, is_flag=True) @click.option('--debug-openai', default=False, is_flag=True) @click.option('--debug-recovery', default=False, is_flag=True) -async def cli(debug=False, debug_asyncio=False, debug_openai=False, debug_recovery=True): +async def cli(debug=False, debug_openai=False, debug_recovery=True): logging.basicConfig(level = logging.DEBUG if debug else logging.INFO) diff --git a/src/dbtest/sentences/create.py b/src/dbtest/sentences/create.py index 0a15b5b..701cf43 100644 --- a/src/dbtest/sentences/create.py +++ b/src/dbtest/sentences/create.py @@ -90,7 +90,6 @@ async def create_random_problem(openai_client: AsyncChatGPTClient=AsyncChatGPTCl except Exception as ex: logging.error(traceback.format_exc()) - if display: print(problem_formatter(responses)) diff --git a/src/dbtest/sentences/models.py b/src/dbtest/sentences/models.py index 1b49982..a8dc595 100644 --- a/src/dbtest/sentences/models.py +++ b/src/dbtest/sentences/models.py @@ -1,6 +1,6 @@ from enum import auto import sqlalchemy -from sqlalchemy import Enum, Table, Column, Integer, String, Boolean, quoted_name +from sqlalchemy import Enum, Column, String, Boolean, quoted_name from sqlalchemy.dialects.postgresql import ENUM as SQLAlchemyEnum from dbtest.database.engine import async_engine