FROM codeflash/codeflash-rl-env:latest

# Checkout to pre-optimization state
RUN cd /workspace/inference && git checkout 7d46e2fbacd49d1b0d31ef1e4fcd8a6e72927505

# Re-install to get correct dependency versions for this commit
RUN pip install --no-cache-dir -e . 2>/dev/null || true
RUN pip install --no-cache-dir codeflash==0.20.5

WORKDIR /workspace/inference
