Merge pull request #2216 from nd-02110114/fix-small-bug

Fix __str__ method for Featurizer
deepchem · Oct 14, 2020 · ae12a7e · ae12a7e
2 parents 952375f + 707a7b5
commit ae12a7e
Showing 1 changed file with 12 additions and 11 deletions.
diff --git a/deepchem/feat/base_classes.py b/deepchem/feat/base_classes.py
@@ -134,18 +134,19 @@ def __str__(self) -> str:
 
     override_args_info = ''
     for arg_name, default in zip(args_names, args_default_values):
-      arg_value = self.__dict__[arg_name]
-      # validation
-      # skip list
-      if isinstance(arg_value, list):
-        continue
-      if isinstance(arg_value, str):
-        # skip path string
-        if "\\/." in arg_value or "/" in arg_value or '.' in arg_value:
+      if arg_name in self.__dict__:
+        arg_value = self.__dict__[arg_name]
+        # validation
+        # skip list
+        if isinstance(arg_value, list):
           continue
-      # main logic
-      if default != arg_value:
-        override_args_info += '_' + arg_name + '_' + str(arg_value)
+        if isinstance(arg_value, str):
+          # skip path string
+          if "\\/." in arg_value or "/" in arg_value or '.' in arg_value:
+            continue
+        # main logic
+        if default != arg_value:
+          override_args_info += '_' + arg_name + '_' + str(arg_value)
     return self.__class__.__name__ + override_args_info