From 1114b210f9457f815ff8afce2dd543c27e123791 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gerg=C5=91=20Jedlicska?= Date: Sat, 13 Feb 2021 14:06:32 +0100 Subject: [PATCH] refactor(objects module def): remove redundant subclass discovery code With the automated subclass registry, there is no need for the magic module lookup in objects __init__ --- speckle/objects/__init__.py | 15 +++------------ 1 file changed, 3 insertions(+), 12 deletions(-) diff --git a/speckle/objects/__init__.py b/speckle/objects/__init__.py index b5fc3f9..fa648cf 100644 --- a/speckle/objects/__init__.py +++ b/speckle/objects/__init__.py @@ -1,14 +1,5 @@ -from pathlib import Path -import sys -import inspect -import pkgutil -from importlib import import_module -from .base import Base +"""Builtin Speckle object kit.""" +from speckle.objects.base import Base -for (_, name, _) in pkgutil.iter_modules([Path(__file__).parent]): - imported_module = import_module("." + name, package=__name__) - classes = inspect.getmembers(imported_module, inspect.isclass) - for c in classes: - if issubclass(c[1], Base): - setattr(sys.modules[__name__], c[0], c[1]) +__all__ = ["Base"]