1from inspect import isclass
2from pkgutil import iter_modules
3from pathlib import Path
4from importlib import import_module
5
6# iterate through the modules in the current package
7package_dir = Path(__file__).resolve().parent
8for (_, module_name, _) in iter_modules([package_dir]):
9
10 # import the module and iterate through its attributes
11 module = import_module(f"{__name__}.{module_name}")
12 for attribute_name in dir(module):
13 attribute = getattr(module, attribute_name)
14
15 if isclass(attribute):
16 # Add the class to this package's variables
17 globals()[attribute_name] = attribute
18