From 2a9070f1ad4723ab2792bdda21d83f98573c5717 Mon Sep 17 00:00:00 2001
From: Kevin M Jablonka <32935233+kjappelbaum@users.noreply.github.com>
Date: Sun, 18 Dec 2022 00:21:11 +0100
Subject: [PATCH] pretty print fails for the `RDKitAdaptor` Featurizers. (#413)

Fixes #375
---
 src/mofdscribe/featurizers/bu/rdkitadaptor.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/mofdscribe/featurizers/bu/rdkitadaptor.py b/src/mofdscribe/featurizers/bu/rdkitadaptor.py
index 4d2cc51..370f77e 100644
--- 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