Refactor Formatters, use functional approach

1 files changed, 4 insertions, 18 deletions

diff --git a/robusta_krr/formatters/pprint.py b/robusta_krr/formatters/pprint.py
index bdfcc4c..9be637c 100644
--- a/robusta_krr/formatters/pprint.py
+++ b/robusta_krr/formatters/pprint.py
@@ -1,23 +1,9 @@
-from __future__ import annotations
-
 from pprint import pformat
 
-from robusta_krr.core.abstract.formatters import BaseFormatter
+from robusta_krr.core.abstract import formatters
 from robusta_krr.core.models.result import Result
 
 
-class PPrintFormatter(BaseFormatter):
-    """Formatter for object output with python's pprint module."""
-
-    __display_name__ = "pprint"
-
-    def format(self, result: Result) -> str:
-        """Format the result using pprint.pformat(...)
-
-        :param result: The results to format.
-        :type result: :class:`core.result.Result`
-        :returns: The formatted results.
-        :rtype: str
-        """
-
-        return pformat(result.dict())
+@formatters.register()
+def pprint(result: Result) -> str:
+    return pformat(result.dict())