-
Notifications
You must be signed in to change notification settings - Fork 150
Expand file tree
/
Copy pathsetup.py
More file actions
29 lines (27 loc) · 948 Bytes
/
setup.py
File metadata and controls
29 lines (27 loc) · 948 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from setuptools import setup, find_packages
# Read dependencies from requirements.txt
with open("requirements.txt") as f:
requirements = f.read().splitlines()
setup(
name="open-unlearning",
version="0.1.0",
author="Vineeth Dorna, Anmol Mekala",
description="A library for machine unlearning in LLMs.",
long_description=open("README.md").read(),
long_description_content_type="text/markdown",
url="https://github.com/locuslab/open-unlearning",
license="MIT",
packages=find_packages(),
install_requires=requirements, # Uses requirements.txt
extras_require={
"lm-eval": [
"lm-eval==0.4.11",
], # Install using `pip install ".[lm-eval]"`
"dev": [
"pre-commit==4.0.1",
"ruff==0.6.9",
], # Install using `pip install ".[dev]"`
},
python_requires=">=3.11",
)