fix type

OpenAdaptAI · Apr 19, 2023 · e7ba06c · e7ba06c
1 parent da13e5f
commit e7ba06c
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/puterbot/strategies/llm_ocr_demo.py b/puterbot/strategies/llm_ocr_demo.py
@@ -7,10 +7,8 @@
 """
 
 
-import mss.base
-
 from puterbot.events import get_events
-from puterbot.models import Recording
+from puterbot.models import Recording, Screenshot
 from puterbot.strategies.base import BaseReplayStrategy
 from puterbot.strategies.llm_mixin import LLMReplayStrategyMixin
 from puterbot.strategies.ocr_mixin import OCRReplayStrategyMixin
@@ -30,7 +28,7 @@ def __init__(
 
     def get_next_input_event(
         self,
-        screenshot: mss.base.ScreenShot,
+        screenshot: Screenshot,
     ):
         text = self.get_text(screenshot)