Update _calculate.py to fix square root computations.

mhostetter · Nov 22, 2024 · 49cacb9 · 49cacb9
1 parent c945356
commit 49cacb9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/galois/_domains/_calculate.py b/src/galois/_domains/_calculate.py
@@ -783,11 +783,12 @@ def implementation(self, a: Array) -> Array:
             )
 
         p = self.field.characteristic
+        q = self.field.order
 
-        if p % 4 == 3:
+        if q % 4 == 3:
             roots = a ** ((self.field.order + 1) // 4)
 
-        elif p % 8 == 5:
+        elif q % 8 == 5:
             d = a ** ((self.field.order - 1) // 4)
             roots = self.field.Zeros(a.shape)