matches with numba
@@ -76,7 +76,7 @@ def compute_viral_load(t, time_start, time_recovered, time_dead, frac_time,
return load
-# @nb.njit( (nbfloat[:], nbfloat[:], nbbool[:], nbbool[:], nbfloat, nbfloat[:], nbbool[:], nbbool[:], nbbool[:], nbfloat, nbfloat, nbfloat, nbfloat[:]), cache=True, parallel=safe_parallel)
+@nb.njit( (nbfloat[:], nbfloat[:], nbbool[:], nbbool[:], nbfloat, nbfloat[:], nbbool[:], nbbool[:], nbbool[:], nbfloat, nbfloat, nbfloat, nbfloat[:]), cache=True, parallel=safe_parallel)
def compute_trans_sus(rel_trans, rel_sus, inf, sus, beta_layer, viral_load, symp, diag, quar, asymp_factor, iso_factor, quar_factor, immunity_factors): # pragma: no cover
''' Calculate relative transmissibility and susceptibility '''
f_asymp = symp + ~symp * asymp_factor # Asymptomatic factor, changes e.g. [0,1] with a factor of 0.8 to [0.8,1.0]
@@ -90,7 +90,7 @@ def compute_trans_sus(rel_trans, rel_sus, inf, sus, beta_layer,
return rel_trans, rel_sus
-# @nb.njit( (nbfloat, nbint[:], nbint[:], nbfloat[:], nbfloat[:], nbfloat[:]), cache=cache, parallel=rand_parallel)
+@nb.njit( (nbfloat, nbint[:], nbint[:], nbfloat[:], nbfloat[:], nbfloat[:]), cache=cache, parallel=rand_parallel)
def compute_infections(beta, sources, targets, layer_betas, rel_trans, rel_sus): # pragma: no cover
'''
Compute who infects whom
