[Misc] Move numba installation to requirements.txt (#948)

Signed-off-by: Pooya Moradi <pooyam@google.com>
Co-authored-by: Pooya Moradi <pooyam@google.com>

Author: py4

docker/Dockerfile

@@ -28,7 +28,6 @@ RUN python3 -m pip install --no-cache-dir \
     git+https://github.com/EleutherAI/lm-evaluation-harness.git@206b7722158f58c35b7ffcd53b035fdbdda5126d#egg=lm-eval[api] \
     pytest-cov \
     tblib \
-    numba
 
 # Install tpu_inference
 WORKDIR /workspace/tpu_inference

requirements.txt

@@ -15,3 +15,4 @@ qwix==0.1.1
 torchvision==0.23.0
 pathwaysutils
 parameterized
+numba==0.62.1