docstring for the win

datamol-io · Jun 22, 2023 · e01067a · e01067a
1 parent 1e1b90a
commit e01067a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/datamol/viz/utils.py b/datamol/viz/utils.py
@@ -137,8 +137,10 @@ def image_to_file(
 def to_rdkit_color(color: Optional[DatamolColor]) -> Optional[RDKitColor]:
     """If required convert a datamol color (rgb, rgba or hex string) to an RDKit
     color (rgb or rgba).
+
+    Args:
+        color: A datamol color: hex, rgb, rgba or None.
     """
     if isinstance(color, str):
         return mcolors.to_rgba(color)  # type: ignore
-    else:
-        return color
+    return color