Make "poly" display mode more efficient for prime fields

mhostetter · Mar 2, 2022 · 7ffd551 · 7ffd551
1 parent 3a0c668
commit 7ffd551
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/galois/_fields/_main.py b/galois/_fields/_main.py
@@ -443,9 +443,13 @@ def arithmetic_table(
  ###############################################################################
 
  def _formatter(cls, array):
+ """
+ Returns a NumPy printoptions "formatter" dictionary.
+ """
  # pylint: disable=attribute-defined-outside-init
  formatter = {}
- if cls.display_mode == "poly":
+
+ if cls.display_mode == "poly" and cls.is_extension_field:
  formatter["int"] = cls._print_poly
  formatter["object"] = cls._print_poly
  elif cls.display_mode == "power":
@@ -454,6 +458,7 @@ def _formatter(cls, array):
  formatter["object"] = cls._print_power
  elif array.dtype == np.object_:
  formatter["object"] = cls._print_int
+
  return formatter
 
  def _print_int(cls, element): # pylint: disable=no-self-use