fix issues

awkrail · Oct 21, 2024 · 4a35ad7 · 4a35ad7
1 parent eea74a1
commit 4a35ad7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git 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
@@ -17,4 +17,7 @@ ignore_missing_imports = True
 ignore_missing_imports = True
 
 [mypy-msclap.*]
+ignore_missing_imports = True
+
+[mypy-timm.*]
 ignore_missing_imports = True