Move project description from setup.py to pyproject.toml (#21)

Also applied ruff

Author: nandor

pyproject.toml

@@ -1,9 +1,13 @@
-[tool.poetry]
-authors = ["Perplexity AI"]
-description = ""
+[project]
 name = "pplx-kernels"
-readme = "README.md"
 version = "0.0.1"
+description = "Perplexity CUDA Kernels"
+readme = "README.md"
+requires-python = ">=3.12"
+
+[build-system]
+requires = ["setuptools>=61.0", "wheel", "torch"]
+build-backend = "setuptools.build_meta"
 
 [tool.ruff]
 line-length = 88

setup.py

@@ -6,8 +6,6 @@
 from setuptools.command.build import build as _build
 from setuptools.command.build_ext import build_ext
 
-VERSION = "0.0.1"
-
 
 def _get_torch_cmake_prefix_path() -> str:
     import torch
@@ -88,9 +86,6 @@ def run(self) -> None:
 ]
 
 setup(
-    name="pplx-kernels",
-    version=VERSION,
-    description="Perplexity Kernels",
     packages=find_packages(where="src"),
     package_dir={"": "src"},
     package_data={
@@ -103,8 +98,6 @@ def run(self) -> None:
     },
     options={"bdist_wheel": {"py_limited_api": "cp39"}},
     zip_safe=False,
-    install_requires=["torch"],
-    python_requires=">=3.10",
     ext_modules=extensions,
     include_package_data=True,
 )

src/pplx_kernels/all_to_all.py

@@ -1,6 +1,7 @@
 # pyright: reportCallIssue=false
 
-from typing import Any, Callable
+from collections.abc import Callable
+from typing import Any
 
 import torch
 

tests/distributed_utils.py

@@ -1,7 +1,8 @@
 import dataclasses
 import logging
 import os
-from typing import Callable, Concatenate, ParamSpec
+from collections.abc import Callable
+from typing import Concatenate, ParamSpec
 
 import pytest
 import torch

tests/test_all_to_all.py

@@ -130,7 +130,7 @@ def _do_test_all_to_all(
                 logger.debug(
                     "    x[%d] -> %s",
                     token_idx,
-                    list(zip(indices, weights)),
+                    list(zip(indices, weights, strict=False)),
                 )
             for token_idx in range(rd.num_tokens):
                 logger.debug("    x[%d]=%s", token_idx, _str_1d_tensor(rd.x[token_idx]))