From 4a35ad759ab44583a6a24a9aa2e2fd82ac9c6dc2 Mon Sep 17 00:00:00 2001
From: awkrail <taichitary@gmail.com>
Date: Mon, 21 Oct 2024 11:49:08 +0900
Subject: [PATCH] fix issues

---
 firefly/base_encoder.py | 4 +++-
 mypy.ini                | 3 +++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/firefly/base_encoder.py b/firefly/base_encoder.py
index 57e35e7..1b6ef46 100644
--- a/firefly/base_encoder.py
+++ b/firefly/base_encoder.py
@@ -1,7 +1,9 @@
 import abc
 import torch
 
-from typing import Tuple
+from typing import List
+
+from firefly.frame_extractor.frame import VideoFrame, AudioFrame
 
 class BaseVisionEncoder(metaclass=abc.ABCMeta):
     @abc.abstractmethod
diff --git a/mypy.ini b/mypy.ini
index 6ed21c0..a261c1f 100644
--- a/mypy.ini
+++ b/mypy.ini
@@ -17,4 +17,7 @@ ignore_missing_imports = True
 ignore_missing_imports = True
 
 [mypy-msclap.*]
+ignore_missing_imports = True
+
+[mypy-timm.*]
 ignore_missing_imports = True
\ No newline at end of file