feat: pr feedback json kwargs

Signed-off-by: jaapschoutenalliander <[email protected]>

Author: jaapschoutenalliander

src/power_grid_model_ds/_core/model/grids/base.py

@@ -439,9 +439,16 @@ def from_txt_file(cls, txt_file_path: Path):
             txt_lines = f.readlines()
         return TextSource(grid_class=cls).load_from_txt(*txt_lines)
 
-    def to_json(self, path: Path) -> Path:
-        """Serialize the grid to JSON format."""
-        return _save_grid_to_json(grid=self, path=path)
+    def to_json(self, path: Path, **kwargs) -> Path:
+        """Serialize the grid to JSON format.
+
+        Args:
+            path: Destination file path to write JSON to.
+            **kwargs: Additional keyword arguments forwarded to ``json.dump``
+        Returns:
+            Path: The path where the file was saved.
+        """
+        return _save_grid_to_json(grid=self, path=path, **kwargs)
 
     @classmethod
     def from_json(cls: Type[Self], path: Path) -> Self:

src/power_grid_model_ds/_core/utils/serialization.py

@@ -8,7 +8,7 @@
 import json
 import logging
 from pathlib import Path
-from typing import TYPE_CHECKING, Dict, Optional, Type, TypeVar
+from typing import TYPE_CHECKING, Dict, Type, TypeVar
 
 from power_grid_model_ds._core.model.arrays.base.array import FancyArray
 
@@ -52,14 +52,15 @@ def _restore_grid_values(grid, json_data: Dict) -> None:
 def _save_grid_to_json(
     grid,
     path: Path,
-    indent: Optional[int] = None,
+    **kwargs,
 ) -> Path:
     """Save a Grid object to JSON format using power-grid-model serialization with extensions support.
 
     Args:
         grid: The Grid object to serialize
         path: The file path to save to
-        indent: JSON indentation (None for compact, positive int for indentation)
+        **kwargs: Keyword arguments forwarded to json.dump (for example, indent, sort_keys,
+            ensure_ascii, etc.).
     Returns:
         Path: The path where the file was saved
     """
@@ -87,7 +88,7 @@ def _save_grid_to_json(
 
     # Write to file
     with open(path, "w", encoding="utf-8") as f:
-        json.dump(serialized_data, f, indent=indent if indent and indent > 0 else None)
+        json.dump(serialized_data, f, **kwargs)
 
     return path