Derivative Container via Vertex AI Workbench unable to find '/app/run_module.py'

[Alif Nur Patriya Sussardi]

I am using Vertex AI Workbench usually to do simple stuff like knn or lgbm modellling on a scheduled time. 

Yesterday, I am trying to finetune BERT for classifier, which cant be done using existing Python 3 environment that's already available from the Notebook Executor. 

Therefore, I build a docker Image to run this. This is my dockerfile: 

```

# Dockerfile
FROM us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-4.py310:latest

WORKDIR /app

COPY requirements.txt .

RUN pip install --no-cache-dir --upgrade -r requirements.txt

``` 

The container is built alright and pushed to artifact registry. 

I use this container to run the job from workbench jupyter notebook executor, but got persisting error that I cant figured how to fix it. I think it has something to do with how notebook executor use derivative container. 

This is the error: 

python3: can't open file '/app/run_module.py': [Errno 2] No such file or directory I believe there's something I could do with the dockerfile that fix this. 
Could the team can suggest me how to fix this?