From 0cd10c78f410adb32abe736430a6441dfc0f7384 Mon Sep 17 00:00:00 2001 From: AgentDS Date: Sun, 21 Jan 2024 17:27:30 -0500 Subject: [PATCH] add requirement for seaborn --- requirements.txt | 1 + setup.py | 51 ++++++++++++++++++++++++------------------------ 2 files changed, 27 insertions(+), 25 deletions(-) diff --git a/requirements.txt b/requirements.txt index 14f73006..0649f5bf 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,6 +4,7 @@ pynvml tqdm scikit-learn munch +seaborn torch>=1.7.1 torchvision>=0.8.2 \ No newline at end of file diff --git a/setup.py b/setup.py index 59c14a6a..139547de 100644 --- a/setup.py +++ b/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", +)