diff --git a/cogs/message_listener/__init__.py b/cogs/message_listener/__init__.py index 18c569d..198f658 100644 --- a/cogs/message_listener/__init__.py +++ b/cogs/message_listener/__init__.py @@ -12,8 +12,8 @@ from sqlalchemy.orm import Session # pylint: disable=import-error from classes.submission import SubmissionState -from main import submission_objects from database_setup.sql_data_classes import Category, Guild, engine +from main import submission_objects class SubmissionListener(commands.Cog, name="Message Listener"): diff --git a/cogs/submission/submission_views.py b/cogs/submission/submission_views.py index 3c38470..9d6807c 100644 --- a/cogs/submission/submission_views.py +++ b/cogs/submission/submission_views.py @@ -7,8 +7,8 @@ from sqlalchemy.orm import Session, aliased # pylint: disable=import-error from classes.submission import Submission, SubmissionState -from main import submission_objects from database_setup.sql_data_classes import Category, CategoryGroup, Guild, engine +from main import submission_objects from .verification_view import VerificationView