From d361a22df5d81eed72cf564a08c6f0817a1d2f30 Mon Sep 17 00:00:00 2001
From: KevinCawley <72036796+KevinCawley@users.noreply.github.com>
Date: Thu, 20 May 2021 14:27:07 -0500
Subject: [PATCH] Fix numba deprecation (#1571)

* Fix jitclass

* update numba version to not be fixed

* Move pickle5 to pip section

* Move pickle5 back to conda-forge

* Pin numba version to latest stable

* Leading 0
---
 tardis/montecarlo/montecarlo_numba/formal_integral.py | 3 ++-
 tardis_env3.yml                                       | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/tardis/montecarlo/montecarlo_numba/formal_integral.py b/tardis/montecarlo/montecarlo_numba/formal_integral.py
index b4b791b1b37..0c7d3c6627c 100644
--- a/tardis/montecarlo/montecarlo_numba/formal_integral.py
+++ b/tardis/montecarlo/montecarlo_numba/formal_integral.py
@@ -6,7 +6,8 @@
 from scipy.interpolate import interp1d
 from astropy import units as u
 from tardis import constants as const
-from numba import njit, char, float64, int64, jitclass, typeof, byte, prange
+from numba import njit, char, float64, int64, typeof, byte, prange
+from numba.experimental import jitclass
 import pdb
 
 from tardis.montecarlo.montecarlo_numba.numba_config import SIGMA_THOMSON
diff --git a/tardis_env3.yml b/tardis_env3.yml
index a10de0ac238..6ffd66c46da 100644
--- a/tardis_env3.yml
+++ b/tardis_env3.yml
@@ -10,7 +10,7 @@ dependencies:
   - scipy=1.5
   - pandas=1.0
   - astropy=3
-  - numba=0.50
+  - numba=0.53.1
   - numexpr
 
   # Plasma