adding link back to constructing class

IDAES · May 21, 2024 · e552d42 · e552d42
1 parent 3cf3a84
commit e552d42
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/idaes/core/base/process_block.py b/idaes/core/base/process_block.py
@@ -114,6 +114,8 @@ def __init__(self, *args, **kwargs):
         dct["__process_block__"] = "indexed"
         # provide function ``base_class_module()`` to get unit module, for visualizer
         dct["base_class_module"] = lambda mcs: bases[0].__module__
+        # provide function ``process_block_class()`` to get the constructing class
+        dct["process_block_class"] = lambda mcs: bases[0]
         return type.__new__(mcs, name, bases, dct)
 
 
@@ -130,6 +132,8 @@ def __init__(self, *args, **kwargs):
         dct["__process_block__"] = "scalar"
         # provide function ``base_class_module()`` to get unit module, for visualizer
         dct["base_class_module"] = lambda mcs: bases[0].__module__
+        # provide function ``process_block_class()`` to get the constructing class
+        dct["process_block_class"] = lambda mcs: bases[1]
         return type.__new__(mcs, name, bases, dct)