fix python3.10 error (#84)

Signed-off-by: Qianyu Wang <patrickwangqy@outlook.com>

Signed-off-by: Qianyu Wang <patrickwangqy@outlook.com>

datasets/OUMVLP/rearrange_OUMVLP.py

@@ -2,6 +2,7 @@ import argparse
 import os
 import shutil
 from pathlib import Path
+from typing import Tuple
 
 from tqdm import tqdm
 
@@ -9,7 +10,7 @@ from tqdm import tqdm
 TOTAL_SUBJECTS = 10307
 
 
-def sanitize(name: str) -> (str, str):
+def sanitize(name: str) -> Tuple[str, str]:
     return name.split('_')[1].split('-')
 
 

opengait/modeling/backbones/__init__.py

@@ -5,13 +5,13 @@ from importlib import import_module
 
 # iterate through the modules in the current package
 package_dir = Path(__file__).resolve().parent
-for (_, module_name, _) in iter_modules([package_dir]):
+for (_, module_name, _) in iter_modules([str(package_dir)]):
 
     # import the module and iterate through its attributes
     module = import_module(f"{__name__}.{module_name}")
     for attribute_name in dir(module):
         attribute = getattr(module, attribute_name)
 
-        if isclass(attribute):            
+        if isclass(attribute):
             # Add the class to this package's variables
             globals()[attribute_name] = attribute

opengait/modeling/losses/__init__.py

@@ -5,13 +5,13 @@ from importlib import import_module
 
 # iterate through the modules in the current package
 package_dir = Path(__file__).resolve().parent
-for (_, module_name, _) in iter_modules([package_dir]):
+for (_, module_name, _) in iter_modules([str(package_dir)]):
 
     # import the module and iterate through its attributes
     module = import_module(f"{__name__}.{module_name}")
     for attribute_name in dir(module):
         attribute = getattr(module, attribute_name)
 
-        if isclass(attribute):            
+        if isclass(attribute):
             # Add the class to this package's variables
             globals()[attribute_name] = attribute

opengait/modeling/models/__init__.py

@@ -5,13 +5,13 @@ from importlib import import_module
 
 # iterate through the modules in the current package
 package_dir = Path(__file__).resolve().parent
-for (_, module_name, _) in iter_modules([package_dir]):
+for (_, module_name, _) in iter_modules([str(package_dir)]):
 
     # import the module and iterate through its attributes
     module = import_module(f"{__name__}.{module_name}")
     for attribute_name in dir(module):
         attribute = getattr(module, attribute_name)
 
-        if isclass(attribute):            
+        if isclass(attribute):
             # Add the class to this package's variables
             globals()[attribute_name] = attribute