pretty print fails for the RDKitAdaptor Featurizers. (#413)

Fixes #375
kjappelbaum · Dec 17, 2022 · 2a9070f · 2a9070f
1 parent 8d57d52
commit 2a9070f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/mofdscribe/featurizers/bu/rdkitadaptor.py b/src/mofdscribe/featurizers/bu/rdkitadaptor.py
@@ -50,6 +50,13 @@ def __init__(
         self._local_env_strategy = local_env_strategy
         self._force_sanitize = force_sanitize
 
+    def __repr__(self) -> str:
+        return "RDKitAdaptor(featurizer={}, feature_labels={})".format(
+            self._featurizer, self._feature_labels)
+
+    def _repr_mimebundle_(self, include=None, exclude=None):
+        return self.__repr__()
+
     def feature_labels(self) -> List[str]:
         return self._feature_labels