.github/workflows/train.yml

@@ -15,12 +15,12 @@ jobs:
 
       - name: Use Docker action
         id: train
-        uses: ./train-action/
+        uses: ./
         with:
           epochs: 1
 
       - name: Archive
-        uses: actions/upload-artifact@v3
+        uses: action/upload-artifact@v3
         with:
           name: model
           path: model.keras

train.dockerfile

@@ -14,4 +14,4 @@ RUN wget https://bendun.students.wmi.amu.edu.pl/stop_times.categories.tsv
 
 VOLUME /github/workspace/
 
-ENTRYPOINT ["/ium/train-action/entrypoint.sh"]
+ENTRYPOINT ["/ium/train-entrypoint.sh"]

train.yml

@@ -7,6 +7,6 @@ inputs:
     default: '2'
 runs:
   using: 'docker'
-  image: 'Dockerfile'
+  image: 'train.dockerfile'
   args:
     - ${{ inputs.epochs }}