Update dockerfile

Dockerfile

@@ -1,7 +1,10 @@
-FROM ubuntu
+FROM ubuntu:22.04
+
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install --no-install-recommends --assume-yes \
+    pip iputils-ping curl wget wkhtmltopdf
 
-RUN apt-get update -y && apt-get upgrade -y && apt-get install python3 pip iputils-ping curl wget -y
 COPY requirements.txt /home/requirements.txt
 RUN pip3 install -r /home/requirements.txt
+RUN ln -s /usr/bin/python3 /usr/bin/python
 
 WORKDIR /flcore

flcore/report/generate_report.py

@@ -20,7 +20,7 @@ def generate_report(experiment_path: str):
     os.makedirs(fig_dir, exist_ok=True)
 
     # Copy images dir to the experiment dir
-    os.system(f'cp -r flcore/report/images/ {fig_dir}')
+    os.system(f'cp -r flcore/report/images/* {fig_dir}')
 
     # Read the CSV file
     df = pd.read_csv(results_file)

flcore/report/template.html

@@ -77,7 +77,7 @@ <h5 class="card-title">{{ key }}</h5>
                 {{ value.table }}
 
                 {%- for plot in value.plot %}
-                <img src="{{ plot }}" width="33%">
+                <img src="{{ plot }}" width="31%">
                 {%- endfor %}
 
             </p>