self.__class__.__name__ -> type(self).__name__

Author: janosh

crystal_toolkit/components/transformations/core.py

@@ -24,7 +24,7 @@
 class TransformationComponent(MPComponent):
     def __init__(self, input_structure_component_id: str, *args, **kwargs):
 
-        if self.__class__.__name__ != f"{self.transformation.__name__}Component":
+        if type(self).__name__ != f"{self.transformation.__name__}Component":
             # sanity check, enforcing conventions
             raise NameError(
                 f"Class has to be named corresponding to the underlying "
@@ -230,7 +230,7 @@ def update_preview(transformation_data, input_structure):
         )
         @cache.memoize(
             timeout=60 * 60 * 24,
-            make_name=lambda x: f"{self.__class__.__name__}_{x}_cached",
+            make_name=lambda x: f"{type(self).__name__}_{x}_cached",
         )
         def update_transformation(enabled, states):
 

crystal_toolkit/core/mpcomponent.py

@@ -207,7 +207,7 @@ def __init__(
         if id is None:
             # TODO: this could lead to duplicate ids and an error, but if
             # setting random ids, this could also lead to undefined behavior
-            id = f"{CT_NAMESPACE}{self.__class__.__name__}"
+            id = f"{CT_NAMESPACE}{type(self).__name__}"
         elif not id.startswith(CT_NAMESPACE):
             id = f"{CT_NAMESPACE}{id}"
         MPComponent._all_id_basenames.add(id)
@@ -228,7 +228,7 @@ def __init__(
             # callbacks generated as final step by crystal_toolkit_layout()
             self._callbacks_to_generate.add(self)
 
-        self.logger = logging.getLogger(self.__class__.__name__)
+        self.logger = logging.getLogger(type(self).__name__)
 
     def id(
         self,
@@ -360,7 +360,7 @@ def all_ids(self) -> List[str]:
         )
 
     def __repr__(self):
-        return f"{self.id()}<{self.__class__.__name__}>"
+        return f"{self.id()}<{type(self).__name__}>"
 
     def __str__(self):
         ids = "\n".join(
@@ -371,7 +371,7 @@ def __str__(self):
             [f"* {layout}  " for layout in sorted(self._sub_layouts.keys())]
         )
 
-        return f"""{self.id()}<{self.__class__.__name__}>  \n
+        return f"""{self.id()}<{type(self).__name__}>  \n
 IDs:  \n{ids}  \n
 Stores:  \n{stores}  \n
 Sub-layouts:  \n{layouts}"""
@@ -764,7 +764,7 @@ def reconstruct_kwargs_from_state(self, state=None, kwarg_labels=None) -> Dict:
                 kwargs[k] = v.tolist()
 
         if SETTINGS.DEBUG_MODE:
-            print(self.__class__.__name__, "kwargs", kwargs)
+            print(type(self).__name__, "kwargs", kwargs)
 
         return kwargs