Fix types

Author: cffls

pycardano/address.py

@@ -416,6 +416,7 @@ def save(
         path: str,
         key_type: Optional[str] = None,
         description: Optional[str] = None,
+        **kwargs,
     ):
         """
         Save the Address object to a file.
@@ -427,6 +428,7 @@ def save(
             path (str): The file path to save the object to.
             key_type (str, optional): Not used in this context, but can be included for consistency.
             description (str, optional): Not used in this context, but can be included for consistency.
+            **kwargs: Additional keyword arguments (not used here).
 
         Raises:
             IOError: If the file already exists and is not empty.

pycardano/key.py

@@ -73,7 +73,7 @@ def to_primitive(self) -> bytes:
     def from_primitive(cls: Type["Key"], value: bytes) -> Key:
         return cls(value)
 
-    def to_json(self, **kwargs) -> str:
+    def to_json(self, **kwargs) -> str:  # type: ignore
         """Serialize the key to JSON.
 
         The json output has three fields: "type", "description", and "cborHex".
@@ -90,7 +90,7 @@ def to_json(self, **kwargs) -> str:
         )
 
     @classmethod
-    def from_json(cls: Type[Key], data: str, validate_type=False) -> Key:
+    def from_json(cls: Type[Key], data: str, validate_type=False) -> Key:  # type: ignore
         """Restore a key from a JSON string.
 
         Args:

pycardano/plutus.py

@@ -652,7 +652,7 @@ def _dfs(obj):
 
         return _dfs(self)
 
-    def to_json(self, **kwargs) -> str:
+    def to_json(self, **kwargs) -> str:  # type: ignore
         """Convert to a json string
 
         Args:
@@ -847,7 +847,7 @@ def _dfs(obj):
 
         return _dfs(RawPlutusData.to_primitive(self))
 
-    def to_json(self, **kwargs) -> str:
+    def to_json(self, **kwargs) -> str:  # type: ignore
         """Convert to a json string
 
         Args:

pycardano/serialization.py

@@ -561,7 +561,12 @@ def json_description(self) -> str:
         """
         return self.__class__.__doc__ or "Generated with PyCardano"
 
-    def to_json(self, key_type: Optional[str] = None, description: Optional[str] = None) -> str:
+    def to_json(
+        self,
+        key_type: Optional[str] = None,
+        description: Optional[str] = None,
+        **kwargs,
+    ) -> str:
         """
         Convert the CBORSerializable object to a JSON string containing type, description, and CBOR hex.
 
@@ -570,17 +575,21 @@ def to_json(self, key_type: Optional[str] = None, description: Optional[str] = N
         Args:
             key_type (str): The type to use in the JSON output. Defaults to the class name.
             description (str): The description to use in the JSON output. Defaults to the class docstring.
+            **kwargs: Extra key word arguments to be passed to `json.dumps()`
 
         Returns:
             str: The JSON string representation of the object.
         """
+        if "indent" not in kwargs:
+            kwargs["indent"] = 2
+
         return json.dumps(
             {
                 "type": key_type or self.json_type,
                 "description": description or self.json_description,
                 "cborHex": self.to_cbor_hex(),
             },
-            indent=2,
+            **kwargs,
         )
 
     @classmethod
@@ -613,6 +622,7 @@ def save(
         path: str,
         key_type: Optional[str] = None,
         description: Optional[str] = None,
+        **kwargs,
     ):
         """
         Save the CBORSerializable object to a file in JSON format.
@@ -624,14 +634,15 @@ def save(
             path (str): The file path to save the object to.
             key_type (str, optional): The type to use in the JSON output. Defaults to the class name.
             description (str, optional): The description to use in the JSON output. Defaults to the class docstring.
+            **kwargs: Extra key word arguments to be passed to `json.dumps()`
 
         Raises:
             IOError: If the file already exists and is not empty.
         """
         if os.path.isfile(path) and os.stat(path).st_size > 0:
             raise IOError(f"File {path} already exists!")
         with open(path, "w") as f:
-            f.write(self.to_json(key_type=key_type, description=description))
+            f.write(self.to_json(key_type=key_type, description=description, **kwargs))
 
     @classmethod
     def load(cls, path: str):