# Nasz obraz będzie dzidziczył z obrazu Ubuntu w wersji latest
FROM ubuntu:latest

# Instalujemy niezbędne zależności. Zwróć uwagę na flagę "-y" (assume yes)
RUN apt update && apt install -y figlet

# Stwórzmy w kontenerze (jeśli nie istnieje) katalog /app i przejdźmy do niego (wszystkie kolejne polecenia RUN, CMD, ENTRYPOINT, COPY i ADD będą w nim wykonywane)
WORKDIR /app

RUN apt-get install -y python3
RUN apt-get install -y python3-pip
RUN python3 -m pip install pandas
RUN python3 -m pip install numpy
RUN python3 -m pip install torch
RUN python3 -m pip install torchvision

COPY ./zadanie1.py ./
COPY ./Customers.csv ./
COPY ./train.py ./
#COPY ./test.py ./
#COPY ./cifar_net.pth ./

RUN chmod +r ./Customers.csv
#RUN chmod +x ./zadanie1.py
RUN chmod +x ./train.py
#RUN chmod +x ./test.py

ARG epochs=5
RUN echo $epochs

CMD python3 ./train.py $epochs
#CMD python3 ./test.py