From 52877a4a71264c92a5fd95e47a34a9032cc583cc Mon Sep 17 00:00:00 2001
From: Justin Hoffman <Justin@JustinDeeHoffman.com>
Date: Wed, 11 Dec 2024 22:08:50 -0700
Subject: [PATCH] Add a pyproject.toml file. Remove setup.py

---
 pyproject.toml | 49 +++++++++++++++++++++++++++++++
 setup.py       | 80 --------------------------------------------------
 2 files changed, 49 insertions(+), 80 deletions(-)
 create mode 100644 pyproject.toml
 delete mode 100644 setup.py

diff --git a/pyproject.toml b/pyproject.toml
new file mode 100644
index 0000000..27f1738
--- /dev/null
+++ b/pyproject.toml
@@ -0,0 +1,49 @@
+[build-system]
+requires = ["setuptools>=42", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "neural-amp-modeler"
+version = "0.11.0"
+description = "Neural Amp Modeler"
+authors = [
+    {name = "Steven Atkinson", email = "steven@atkinson.mn"},
+]
+readme = "README.md"
+license = {file = "LICENSE"}
+requires-python = ">=3.8"
+dependencies = [
+    "auraloss==0.3.0",
+    "matplotlib",
+    "pydantic>=2.0.0",
+    "pytorch_lightning",
+    "scipy",
+    "sounddevice",
+    "tensorboard",
+    "torch",
+    "tqdm",
+    "wavio>=0.0.5",
+]
+
+[project.optional-dependencies]
+# Optional features
+transformers-compat = [
+    "transformers>=4",  # Issue-294
+    "numpy<2",  # For older PyTorch versions; user must know when to include
+]
+
+test = [
+    "pytest",
+    "pytest-mock",
+    "requests",
+]
+
+[project.urls]
+homepage = "https://github.com/sdatkinson/"
+
+[project.scripts]
+nam = "nam.cli:nam_gui"
+nam-full = "nam.cli:nam_full"
+
+[tool.setuptools]
+packages = ["nam"]
diff --git a/setup.py b/setup.py
deleted file mode 100644
index 03bb41c..0000000
--- a/setup.py
+++ /dev/null
@@ -1,80 +0,0 @@
-# File: setup.py
-# Created Date: 2020-04-08
-# Author: Steven Atkinson (steven@atkinson.mn)
-
-from distutils.util import convert_path
-from setuptools import setup, find_packages
-
-
-def get_additional_requirements():
-    additional_requirements = []
-    # Issue 294
-    try:
-        import transformers
-
-        # This may not be unnecessarily straict a requirement, but I'd rather
-        # fix this promptly than leave a chance that it wouldn't be fixed
-        # properly.
-        additional_requirements.append("transformers>=4")
-    except ModuleNotFoundError:
-        pass
-
-    # Issue 494
-    def get_numpy_requirement() -> str:
-        need_numpy_1 = True  # Until proven otherwise
-        try:
-            import torch
-
-            version_split = torch.__version__.split(".")
-            major = int(version_split[0])
-            if major >= 2:
-                minor = int(version_split[1])
-                if minor >= 3:  # Hooray, PyTorch 2.3+!
-                    need_numpy_1 = False
-        except ModuleNotFoundError:
-            # Until I see PyTorch 2.3 come out:
-            pass
-        return "numpy<2" if need_numpy_1 else "numpy"
-
-    additional_requirements.append(get_numpy_requirement())
-
-    return additional_requirements
-
-
-main_ns = {}
-ver_path = convert_path("nam/_version.py")
-with open(ver_path) as ver_file:
-    exec(ver_file.read(), main_ns)
-
-requirements = [
-    "auraloss==0.3.0",
-    "matplotlib",
-    "pydantic>=2.0.0",
-    "pytorch_lightning",
-    "scipy",
-    "sounddevice",
-    "tensorboard",
-    "torch",
-    "transformers>=4",  # Issue-294
-    "tqdm",
-    "wavio>=0.0.5",  # Breaking change with older versions
-]
-requirements.extend(get_additional_requirements())
-
-setup(
-    name="neural-amp-modeler",
-    version=main_ns["__version__"],
-    description="Neural amp modeler",
-    author="Steven Atkinson",
-    author_email="steven@atkinson.mn",
-    url="https://github.com/sdatkinson/",
-    install_requires=requirements,
-    packages=find_packages(),
-    include_package_data=True,
-    entry_points={
-        "console_scripts": [
-            "nam = nam.cli:nam_gui",  # GUI trainer
-            "nam-full = nam.cli:nam_full",  # Full-featured trainer
-        ]
-    },
-)