code: import sys
  sys.path.insert(0,"/content/federated-learning/")
  
  import matplotlib
  matplotlib.use('Agg')
  import matplotlib.pyplot as plt
  import copy
  import numpy as np
  from torchvision import datasets, transforms
  import torch
  
  from utils.sampling import mnist_iid, mnist_noniid, cifar_iid
  from utils.options import args_parser
  from models.Update import LocalUpdate
  from models.Nets import MLP, CNNMnist, CNNCifar
  from models.Fed import FedAvg
  from models.test import test_img
  
apis: matplotlib.use