diff --git a/packages/application/__init__.py b/packages/application/__init__.py index e7fc24d04..e0c234713 100755 --- a/packages/application/__init__.py +++ b/packages/application/__init__.py @@ -27,6 +27,7 @@ from .generated_videos import ( from .generation_tasks import ( CreateGenerationTaskCommand, CreateGenerationTaskUseCase, + GetGenerationTaskUseCase, ) from .ingest_jobs import SubmitIngestJobCommand, SubmitIngestJobUseCase from .jobs import ( @@ -63,6 +64,7 @@ __all__ = [ "CreateAssetUseCase", "CreateGenerationTaskCommand", "CreateGenerationTaskUseCase", + "GetGenerationTaskUseCase", "CreateJobCommand", "CreateJobUseCase", "CreateProjectCommand",