Update Dockerfile

Dockerfile

@@ -20,6 +20,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     wget \
     libopenblas-dev
 
+# Create a directory for numba cache
+RUN mkdir -p /tmp/.numba_cache
+
+# Set environment variable for numba cache
+ENV NUMBA_CACHE_DIR=/tmp/.numba_cache
+
 # Create a virtual environment
 RUN python3 -m venv venv