Fixes for changes relating to Numba error types. (#319)

quartical/gains/amplitude/kernel.py

@@ -227,6 +227,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -449,6 +451,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -505,6 +509,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
     param_to_gain = param_to_gain_factory(corr_mode)
 

quartical/gains/complex/diag_kernel.py

@@ -206,6 +206,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -428,6 +430,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -484,6 +488,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
 
     def impl(

quartical/gains/complex/kernel.py

@@ -207,6 +207,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -429,6 +431,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -485,6 +489,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
 
     def impl(

quartical/gains/crosshand_phase/kernel.py

@@ -225,6 +225,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -453,6 +455,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -509,6 +513,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
     param_to_gain = param_to_gain_factory(corr_mode)
 

quartical/gains/delay/kernel.py

@@ -254,6 +254,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -493,6 +495,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -553,6 +557,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
     param_to_gain = param_to_gain_factory(corr_mode)
 

quartical/gains/delay_and_offset/kernel.py

@@ -248,6 +248,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -487,6 +489,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -547,6 +551,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
     param_to_gain = param_to_gain_factory(corr_mode)
 

quartical/gains/leakage/kernel.py

@@ -200,6 +200,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
 
     def impl(

quartical/gains/phase/kernel.py

@@ -232,6 +232,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -460,6 +462,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -516,6 +520,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
     param_to_gain = param_to_gain_factory(corr_mode)
 

quartical/gains/rotation/kernel.py

@@ -228,6 +228,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -463,6 +465,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -521,6 +525,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
 
     if corr_mode.literal_value == 4:

quartical/gains/rotation_measure/kernel.py

@@ -234,6 +234,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -474,6 +476,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -534,6 +538,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
 
     if corr_mode.literal_value == 4:

quartical/gains/tec_and_offset/kernel.py

@@ -250,6 +250,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -490,6 +492,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -550,6 +554,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
     param_to_gain = param_to_gain_factory(corr_mode)