Tidy3dBaseModel._json method that does the custom handling but also takes in kwargs

Author: momchil-flex

tidy3d/components/base.py

@@ -713,6 +713,20 @@ def __eq__(self, other):
     def _json_string(self) -> str:
         """Returns string representation of a :class:`Tidy3dBaseModel`.
 
+        Returns
+        -------
+        str
+            Json-formatted string holding :class:`Tidy3dBaseModel` data.
+        """
+        return self._json()
+
+    def _json(self, indent=INDENT, exclude_unset=False, **kwargs) -> str:
+        """Overwrites the model ``json`` representation with some extra customized handling.
+
+        Parameters
+        -----------
+        **kwargs : kwargs passed to `self.json()`
+
         Returns
         -------
         str
@@ -727,12 +741,9 @@ def make_json_compatible(json_string: str) -> str:
             json_string = json_string.replace("Infinity", '"Infinity"')
             return json_string.replace(tmp_string, '"-Infinity"')
 
-        json_string = self.json(indent=INDENT, exclude_unset=False)
+        json_string = self.json(indent=indent, exclude_unset=exclude_unset, **kwargs)
         json_string = make_json_compatible(json_string)
         return json_string
-        # json_dict = json.loads(json_string)
-
-        # return json.dumps(json_dict)
 
     @classmethod
     def add_type_field(cls) -> None: