fix error when a loss was infinity

app.py

@@ -119,7 +119,7 @@ def get_avg_loss(uids: typing.List[int]) -> typing.Dict[int, float]:
                 continue
             key = f"uid_data.{uid}"
             if key in history:
-                data = [x for x in list(history[key]) if not math.isnan(x)][-MAX_AVG_LOSS_POINTS:]
+                data = [float(x) for x in list(history[key]) if (isinstance(x, float) and not math.isnan(x)) or isinstance(x, int) ][-MAX_AVG_LOSS_POINTS:]
                 if len(data) > 0:
                     result[uid] = statistics.fmean(data)
         if len(result.keys()) == len(uids):