fix(classifier): typing error

williamfzc · Mar 4, 2020 · 3396d40 · 3396d40
1 parent 02c830f
commit 3396d40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stagesepx/classifier/base.py b/stagesepx/classifier/base.py
@@ -90,7 +90,7 @@ def get_offset(self) -> float:
     def get_stage_set(self) -> typing.Set[str]:
         return set(self.get_stage_list())
 
-    def to_dict(self) -> typing.Dict[str, typing.List[SingleClassifierResult]]:
+    def to_dict(self) -> typing.Dict[str, typing.List[typing.List[SingleClassifierResult]]]:
         stage_list = list(self.get_stage_set())
         try:
             int(stage_list[0])