add pyproject.toml

pyproject.toml

@@ -0,0 +1,26 @@
+[build-system]
+requires = ["setuptools", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "repe"
+version = "0.1.0"
+description = "Representation Engineering"
+readme = "README.md"
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+requires-python = ">=3.9"
+dependencies = [
+    "transformers",
+    "accelerate",
+]
+
+[tool.setuptools]
+packages = ["repe"]
+
+[project.urls]
+Homepage = "https://github.com/andyzoujm/representation-engineering"
+Issues = "https://github.com/andyzoujm/representation-engineering/issues"

setup.py

@@ -1,15 +1,3 @@
-from setuptools import setup, find_namespace_packages
+import setuptools
 
-setup(
-    name="repe",
-    version="0.1",
-    description="",
-    packages=find_namespace_packages(),  # This will automatically find packages
-    author="Center for AI Safety",
-    author_email="",
-    url="https://github.com/andyzoujm/representation-engineering",
-    install_requires=[
-        "transformers",
-        "accelerate",
-    ],
-)
+setuptools.setup()