1 files changed, 11 insertions, 0 deletions
diff --git a/src/mlia/core/reporting.py b/src/mlia/core/reporting.py
index 9006602..58a41d3 100644
--- a/src/mlia/core/reporting.py
+++ b/src/mlia/core/reporting.py
@@ -760,3 +760,14 @@ def _apply_format_parameters(
         return report
 
     return wrapper
+
+
+def resolve_output_format(output: Optional[PathOrFileLike]) -> OutputFormat:
+    """Resolve output format based on the output name."""
+    if isinstance(output, (str, Path)):
+        format_from_filename = Path(output).suffix.lstrip(".")
+
+        if format_from_filename in ["json", "csv"]:
+            return cast(OutputFormat, format_from_filename)
+
+    return "plain_text"