replace list[] with List[]

xuqian6078 · Mar 10, 2022 · 632b019 · 632b019
1 parent 1d1d1e9
commit 632b019
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/modules/fuse_data.py b/modules/fuse_data.py
diff --git a/pipeline/eval.py b/pipeline/eval.py
@@ -1,6 +1,8 @@
+from pathlib import Path
+from typing import List
+
 import torch
 import torch.backends.cudnn
-from pathlib import Path
 from tqdm import tqdm
 
 from utils.image_pair import ImagePair
@@ -18,7 +20,7 @@ def __init__(self, net, cudnn: bool = True, half: bool = False, eval: bool = Fal
         self.net = net
 
     @torch.no_grad()
-    def __call__(self, ir_paths: list[Path], vi_paths: list[Path], dst: Path, color: bool = False):
+    def __call__(self, ir_paths: List[Path], vi_paths: List[Path], dst: Path, color: bool = False):
         p_bar = tqdm(enumerate(zip(ir_paths, vi_paths)), total=len(ir_paths))
         for idx, (ir_path, vi_path) in p_bar:
             assert ir_path.stem == vi_path.stem