add requirement for seaborn

requirements.txt

@@ -4,6 +4,7 @@ pynvml
 tqdm
 scikit-learn
 munch
+seaborn
 
 torch>=1.7.1
 torchvision>=0.8.2

setup.py

@@ -12,33 +12,34 @@
     version=fedlab.__version__,
     keywords=["federated learning", "deep learning", "pytorch"],
     author="Dun Zeng, Siqi Liang, Xiangjing Hu",
-    author_email=
-    "zengdun.cs@gmail.com, zszxlsq@gmail.com, starryhu@foxmail.com",
+    author_email="zengdun.cs@gmail.com, zszxlsq@gmail.com, starryhu@foxmail.com",
     maintainer="Dun Zeng, Siqi Liang, Xiangjing Hu",
-    maintainer_email=
-    "zengdun.cs@gmail.com, zszxlsq@gmail.com, starryhu@foxmail.com",
-    description=
-    "A flexible Federated Learning Framework based on PyTorch, simplifying your Federated Learning research.",
+    maintainer_email="zengdun.cs@gmail.com, zszxlsq@gmail.com, starryhu@foxmail.com",
+    description="A flexible Federated Learning Framework based on PyTorch, simplifying your Federated Learning research.",
     license="Apache-2.0 License",
     url="https://github.com/SMILELab-FL/FedLab",
-    packages=find_packages(include=['fedlab','fedlab.*','LICENSE']),
-    install_requires=['torch>=1.7.1',
-                      'torchvision>=0.8.2',
-                      'numpy',
-                      'pandas',
-                      'scikit-learn',
-                      'pynvml',
-                      'tqdm',
-                      'scikit-learn',
-                      'munch'],
-    python_requires='>=3.6',
+    packages=find_packages(include=["fedlab", "fedlab.*", "LICENSE"]),
+    install_requires=[
+        "torch>=1.7.1",
+        "torchvision>=0.8.2",
+        "numpy",
+        "pandas",
+        "scikit-learn",
+        "pynvml",
+        "tqdm",
+        "scikit-learn",
+        "munch",
+        "seaborn",
+    ],
+    python_requires=">=3.6",
     classifiers=[
-        'Topic :: Scientific/Engineering',
-        'Topic :: Scientific/Engineering :: Artificial Intelligence',
-        'Topic :: Software Development',
-        'Topic :: Software Development :: Libraries',
-        'Topic :: Software Development :: Libraries :: Python Modules',
-        'License :: OSI Approved :: Apache Software License',
-        'Programming Language :: Python :: 3'
+        "Topic :: Scientific/Engineering",
+        "Topic :: Scientific/Engineering :: Artificial Intelligence",
+        "Topic :: Software Development",
+        "Topic :: Software Development :: Libraries",
+        "Topic :: Software Development :: Libraries :: Python Modules",
+        "License :: OSI Approved :: Apache Software License",
+        "Programming Language :: Python :: 3",
     ],
-    test_suite="tests.get_tests")
+    test_suite="tests.get_tests",
+)