Update paynt-learner.dockerfile

Add a dedicated WORKDIR to the Dockerfile
randriu · Feb 19, 2024 · b47b455 · b47b455
1 parent 8aab0e1
commit b47b455
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/paynt-learner.dockerfile b/paynt-learner.dockerfile
@@ -4,3 +4,5 @@ FROM $paynt_base
 RUN pip install torch==2.1.*
 
 RUN pip install ipykernel joblib tensorboard==2.15.* einops==0.7.* gym==0.22.* pygame==2.5.*
+
+WORKDIR /opt/learning