From 368b7bb375575f4c6100ea326d5a65741d0d400f Mon Sep 17 00:00:00 2001
From: Paillat <me@paillat.dev>
Date: Thu, 15 Feb 2024 17:53:02 +0100
Subject: [PATCH] Formatting and type hinting

---
 src/engines/BaseEngine.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/engines/BaseEngine.py b/src/engines/BaseEngine.py
index aade7ab..7624c99 100644
--- a/src/engines/BaseEngine.py
+++ b/src/engines/BaseEngine.py
@@ -1,5 +1,7 @@
-from abc import ABC, abstractmethod
 import gradio as gr
+from abc import ABC, abstractmethod
+
+from ..chore import GenerationContext
 
 
 class BaseEngine(ABC):
@@ -8,6 +10,7 @@ class BaseEngine(ABC):
     description: str
 
     def __init__(self):
+        self.ctx: GenerationContext  # This is for type hinting only
         pass
 
     @classmethod