LeoDu 4 лет назад
Родитель
Сommit
b42544826f

+ 1 - 1
backend_refactor/route/analyze.py

@@ -1,5 +1,5 @@
 import uuid, json
-from backend_refactor.service.video.Video3DAnalyzer import Video3DAnalyzer
+from service.video.Video3DAnalyzer import Video3DAnalyzer
 from db import db
 from flask import request
 from db.task import Task

+ 3 - 2
backend_refactor/service/photo/StandingPhotoAnalyzer.py

@@ -1,4 +1,5 @@
-from service.photo import BasePhotoAnalyzer
+from typing import Callable
+from service.photo.BasePhotoAnalyzer import BasePhotoAnalyzer
 
 import cv2
 import matplotlib.pyplot as plt
@@ -22,7 +23,7 @@ class StandingPhotoAnalyzer(BasePhotoAnalyzer):
         assert len(self.args) == 2, "Invalid arguments: expect 2 photo uuids."
         self.args = args
 
-    def analyze(self, callback: function) -> None:
+    def analyze(self, callback: Callable) -> None:
         callback("RUNNING")
         try:
             for i in self.args: