#
# data = [[x, y] for (x, y) in zip((self.client_index.values()),
#                                  self.acc_local_test_data_with_personalized_model_dict.values())]
# table = wandb.Table(data=data, columns=["Client ID", "Personalized Accuracy"])
# wandb.log({"Personalized Accuracy (Personalized Model)": wandb.plot.line(table, "client ID", "Personalized Accuracy",
#                                                                    title="Client ID vs Personalized Accuracy(P)")})
#
# data1 = [[x, y] for (x, y) in
#          zip((self.client_index.values()), self.acc_local_test_data_with_global_model_dict.values())]
# table1 = wandb.Table(data=data1, columns=["Client ID", "Personalized Accuracy"])
# wandb.log({"Personalized Accuracy Global model": wandb.plot.line(table1, "client ID", "Personalized Accuracy",
#                                                                  title="Client ID vs Personalized Accuracy(G)")})
# ## Pi Parameters
# wandb_table1 = wandb.Table(columns=["Client Index", "Pi_Params"])
# for index in range(self.args.client_num_per_round):
#     wandb_table1.add_data(str(index), str(
#         [self.pi_params_dict[index][0], self.pi_params_dict[index][1], self.pi_params_dict[index][2],
#          self.pi_params_dict[index][3], self.pi_params_dict[index][4], self.pi_params_dict[index][5],
#          self.pi_params_dict[index][6], self.pi_params_dict[index][7],
#          self.pi_params_dict[index][8], self.pi_params_dict[index][9], self.pi_params_dict[index][10],
#          self.pi_params_dict[index][11]]))
# wandb.log({"Pi_Params": wandb_table1})
#
# wandb_table2 = wandb.Table(columns=["Client Index", "Personalized Architecture"])
# for index in range(self.args.client_num_per_round):
#     wandb_table2.add_data(str(index), str(self.personalized_architecture[index]))
# wandb.log({"Personalized Architecture": wandb_table2})