From c346e7bba997a26badfa4bbe7fad2c4fcb067a17 Mon Sep 17 00:00:00 2001
From: J08nY
Date: Sun, 12 Feb 2023 19:27:45 +0100
Subject: Save an unnecessary cast in GMPMod.

---
 pyecsca/ec/mod.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/pyecsca/ec/mod.py b/pyecsca/ec/mod.py
index add7581..52f4a80 100644
--- a/pyecsca/ec/mod.py
+++ b/pyecsca/ec/mod.py
@@ -10,7 +10,7 @@ dispatches to the implementation chosen by the runtime configuration of the libr
 import random
 import secrets
 from functools import wraps, lru_cache
-from typing import Type, Dict, Any, Tuple
+from typing import Type, Dict, Any, Tuple, Union
 
 from public import public
 from sympy import Expr, FF
@@ -584,9 +584,9 @@ if has_gmp:
         def __new__(cls, *args, **kwargs):
             return object.__new__(cls)
 
-        def __init__(self, x: int, n: int):
-            self.x = gmpy2.mpz(x % n)
-            self.n = gmpy2.mpz(n)
+        def __init__(self, x: Union[int, gmpy2.mpz], n: Union[int, gmpy2.mpz]):
+            self.n = gmpy2.mpz(n) if not isinstance(n, gmpy2.mpz) else n
+            self.x = gmpy2.mpz(x % self.n) if not isinstance(x, gmpy2.mpz) else x % self.n
 
         def inverse(self) -> "GMPMod":
             if self.x == 0:
-- 
cgit v1.2.3-70-g09d2