pretty print cli output

Author: VipulMascarenhas

ads/cli.py

@@ -8,6 +8,7 @@
 import sys
 
 import fire
+from dataclasses import is_dataclass
 from ads.common import logger
 
 try:
@@ -70,11 +71,32 @@ def _SeparateFlagArgs(args):
 fire.core.parser.SeparateFlagArgs = _SeparateFlagArgs
 
 
+def serialize(data):
+    """Serialize dataclass objects or lists of dataclass objects.
+    Parameters:
+        data: A dataclass object or a list of dataclass objects.
+    Returns:
+        None
+    Prints:
+        The string representation of each dataclass object, or the string representation of any other type of object.
+    """
+    if isinstance(data, list):
+        [print(str(item) if hasattr(item, "__str__") else repr(item)) for item in data]
+    else:
+        print(
+            str(data)
+            if (is_dataclass(data) and hasattr(data, "__str__"))
+            else repr(data)
+        )
+
+
 def cli():
     if len(sys.argv) > 1 and sys.argv[1] == "aqua":
         from ads.aqua.cli import AquaCommand
 
-        fire.Fire(AquaCommand, command=sys.argv[2:], name="ads aqua")
+        fire.Fire(
+            AquaCommand, command=sys.argv[2:], name="ads aqua", serialize=serialize
+        )
     else:
         click_cli()
 

ads/common/serializer.py

@@ -195,7 +195,10 @@ def to_json(
             `None` in case when `uri` provided.
         """
         json_string = json.dumps(
-            self.to_dict(**kwargs), cls=encoder, default=default or self.serialize
+            self.to_dict(**kwargs),
+            cls=encoder,
+            default=default or self.serialize,
+            indent=4,
         )
         if uri:
             self._write_to_file(s=json_string, uri=uri, **kwargs)
@@ -463,9 +466,7 @@ def from_dict(
                 "These fields will be ignored."
             )
 
-        obj = cls(
-            **{key: obj_dict.get(key) for key in allowed_fields}
-        )
+        obj = cls(**{key: obj_dict.get(key) for key in allowed_fields})
 
         for key, value in obj_dict.items():
             if (