Merge pull request #10 from ZhanZiyuan/feature/fix_suffix

Changed `.npy` to `.npz`.
ZhanZiyuan · May 27, 2024 · cf7f59f · cf7f59f
2 parents 4691514 + 6527d5c
commit cf7f59f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pixel_puzzle/pixel_puzzle.py b/pixel_puzzle/pixel_puzzle.py
@@ -36,7 +36,7 @@ def decode_base64(encoded_text: str,
 def shuffle_pixels(origin_image: str,
                    shuffled_image: str,
                    seed: Union[int, None] = None,
-                   index_file: Union[str, Path, None] = Path(__file__).with_suffix(".npy")) -> None:
+                   index_file: Union[str, Path, None] = Path(__file__).with_suffix(".npz")) -> None:
     """
     Shuffle the arrangement of pixels on two dimensions.
     """
@@ -64,7 +64,7 @@ def shuffle_pixels(origin_image: str,
 def recover_pixels(shuffled_image: str,
                    recovered_image: str,
                    seed: Union[int, None] = None,
-                   index_file: Union[str, Path, None] = Path(__file__).with_suffix(".npy")) -> None:
+                   index_file: Union[str, Path, None] = Path(__file__).with_suffix(".npz")) -> None:
     """
     Recover the arrangement of pixels on two dimensions.
     """