fix python3.10 error (#84)

Signed-off-by: Qianyu Wang <patrickwangqy@outlook.com> Signed-off-by: Qianyu Wang <patrickwangqy@outlook.com>
2022-10-19 16:31:21 +08:00
parent 41de2d47cc
commit bb61874e90
4 changed files with 8 additions and 7 deletions
@@ -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