Clean up dockerfile

Dockerfile

@@ -1,18 +1,6 @@
-# FROM python:3-alpine
-
-# RUN mkdir /data
-# VOLUME /data
-
-# EXPOSE 8080
-
-# WORKDIR /data
-
-# CMD ["python", "-m" , "http.server", "8080"]
-
 # Base image to start with
 FROM ubuntu:20.04
-
-# Who maintains this DockerFile
+
 MAINTAINER Bart Cox <[email protected]>
 
 # Run build without interactive dialogue
@@ -28,37 +16,17 @@ RUN apt-get update \
 # Copy the current folder to the working directory
 COPY setup.py ./
 COPY requirements.txt ./
-#COPY fltk ./fltk
-#COPY configs ./configs
 
 # Install all required packages for the generator
 RUN python3 -m pip install -r requirements.txt
 
 ENV GLOO_SOCKET_IFNAME=$NIC
 ENV TP_SOCKET_IFNAME=$NIC
-#ENV GLOO_SOCKET_IFNAME=eth0
-#ENV TP_SOCKET_IFNAME=eth0
-
-#RUN mkdir -p ./data/MNIST
-#COPY ./data/MNIST ../data/MNIST
-#ADD fltk ./fedsim
-#RUN ls -la
-#COPY federated_learning.py ./
-#COPY custom_mnist.py ./
-#RUN ls -la ./fedsim
 
 # Expose the container's port to the host OS
 EXPOSE 5000
 
-# Run command by default for the executing container
-# CMD ["python3", "/opt/Generatrix/rpc_parameter_server.py", "--world_size=2", "--rank=0", "--master_addr=192.168.144.2"]
-
-#CMD python3 /opt/federation-lab/rpc_parameter_server.py --world_size=$WORLD_SIZE --rank=$RANK --master_addr=10.5.0.11
-#CMD python3 /opt/federation-lab/federated_learning.py $RANK $WORLD_SIZE 10.5.0.11
 COPY fltk ./fltk
 COPY configs ./configs
-#CMD python3 ./fltk/__main__.py single configs/experiment.yaml --rank=$RANK
-# CMD python3 -m fltk single configs/experiment_vanilla.yaml --rank=$RANK
-# CMD python3 -m fltk single $EXP_CONFIG --rank=$RANK
-CMD python3 -m fltk remote $EXP_CONFIG $RANK --nic=$NIC --host=$MASTER_HOSTNAME $OPTIONAL_PARAMS
-#CMD python3 setup.py
+
+CMD python3 -m fltk remote $EXP_CONFIG $RANK --nic=$NIC --host=$MASTER_HOSTNAME $OPTIONAL_PARAMS