Hi, getting the following error while running the code in production through step-functions

python: can't open file '/metaflow/driver.py': [Errno 2] No such file or directory

The driver.py is in root folder of the docker container. Dockerfile:

FROM gradle:7.6-jdk-alpine as BUILD_IMAGE
COPY --chown=gradle:gradle . /home/gradle/src
WORKDIR /home/gradle/src
RUN gradle build

FROM openjdk:19-alpine
RUN apk update && apk add bash python3 py3-pip
RUN ln -s /usr/bin/python3 /usr/bin/python
RUN pip3 install metaflow
RUN mkdir /app
COPY --from=BUILD_IMAGE /home/gradle/src/build/libs/ccc-kotlin-standalone.jar /app/ccc.jar
ADD . .

RUN chmod +x BatchInf

why is this error coming up in production but working fine in local?