Skip to content

Commit 3bbe6af

Browse files
benjibcbenjibc
committed
Migrate to pyproject.toml and fix license field for PyPI compatibility
1 parent c6c9e1e commit 3bbe6af

2 files changed

Lines changed: 95 additions & 80 deletions

File tree

‎pyproject.toml‎

Lines changed: 94 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,94 @@
1+
[build-system]
2+
requires = ["setuptools>=61.0", "versioneer[toml]>=0.29"]
3+
build-backend = "setuptools.build_meta"
4+
5+
[project]
6+
name = "reward-kit"
7+
dynamic = ["version"]
8+
authors = [
9+
{name = "Fireworks AI", email = "info@fireworks.ai"},
10+
]
11+
description = "A Python library for defining, testing, and using reward functions"
12+
readme = "README.md"
13+
license = "Apache-2.0"
14+
requires-python = ">=3.10"
15+
classifiers = [
16+
"Programming Language :: Python :: 3",
17+
"Operating System :: OS Independent",
18+
]
19+
dependencies = [
20+
"requests>=2.25.0",
21+
"pydantic>=2.0.0",
22+
"dataclasses-json>=0.5.7",
23+
"fastapi>=0.68.0",
24+
"uvicorn>=0.15.0",
25+
"python-dotenv>=0.19.0",
26+
"openai==1.78.1",
27+
"aiosqlite",
28+
"aiohttp",
29+
"mcp>=1.9.2",
30+
"PyYAML>=5.0",
31+
"datasets==3.6.0",
32+
"fsspec==2025.3.0",
33+
"hydra-core>=1.3.2",
34+
"omegaconf>=2.3.0",
35+
"gymnasium>=0.29.0",
36+
"httpx>=0.24.0",
37+
"fireworks-ai>=0.17.19",
38+
]
39+
40+
[project.urls]
41+
Homepage = "https://github.com/fireworks-ai/reward-kit"
42+
43+
[project.optional-dependencies]
44+
dev = [
45+
"build",
46+
"twine",
47+
"pytest>=6.0.0",
48+
"pytest-asyncio",
49+
"pytest-httpserver",
50+
"werkzeug>=2.0.0",
51+
"black>=21.5b2",
52+
"mypy>=0.812",
53+
"flake8>=3.9.2",
54+
"autopep8>=1.5.0",
55+
"transformers>=4.0.0",
56+
"types-setuptools",
57+
"types-requests",
58+
"types-PyYAML",
59+
"types-docker",
60+
"versioneer>=0.20",
61+
"openai==1.78.1",
62+
"pre-commit",
63+
"e2b>=0.15.0",
64+
"docker==7.1.0",
65+
]
66+
trl = [
67+
"torch>=1.9",
68+
"trl>=0.7.0",
69+
"peft>=0.7.0",
70+
"transformers>=4.0.0",
71+
"accelerate>=0.28.0",
72+
]
73+
deepseek = [
74+
"difflib>=3.0.0",
75+
]
76+
openevals = [
77+
"openevals>=0.1.0",
78+
]
79+
80+
[project.scripts]
81+
fireworks-reward = "reward_kit.cli:main"
82+
reward-kit = "reward_kit.cli:main"
83+
reward-protocol = "reward_kit.cli:main"
84+
85+
[tool.setuptools.packages.find]
86+
include = ["reward_kit*", "reward_protocol*", "development*"]
87+
88+
[tool.versioneer]
89+
VCS = "git"
90+
style = "pep440"
91+
versionfile_source = "reward_kit/_version.py"
92+
versionfile_build = "reward_kit/_version.py"
93+
tag_prefix = "v"
94+
parentdir_prefix = "reward-kit-"

‎setup.py‎

Lines changed: 1 addition & 80 deletions
Original file line numberDiff line numberDiff line change
@@ -1,86 +1,7 @@
1-
from setuptools import find_packages, setup
2-
1+
from setuptools import setup
32
import versioneer
43

54
setup(
6-
name="reward-kit",
75
version=versioneer.get_version(),
86
cmdclass=versioneer.get_cmdclass(),
9-
packages=find_packages(include=["reward_kit*", "reward_protocol*", "development*"]),
10-
install_requires=[
11-
"requests>=2.25.0",
12-
"pydantic>=2.0.0",
13-
"dataclasses-json>=0.5.7",
14-
"fastapi>=0.68.0",
15-
"uvicorn>=0.15.0",
16-
"python-dotenv>=0.19.0",
17-
"openai==1.78.1",
18-
"aiosqlite",
19-
"aiohttp",
20-
"mcp>=1.9.2",
21-
"PyYAML>=5.0", # Added PyYAML
22-
"datasets==3.6.0", # For dataset loading functionality
23-
"fsspec==2025.3.0", # For filesystem interactions, pinned for dataset compatibility
24-
"hydra-core>=1.3.2", # For configuration management
25-
"omegaconf>=2.3.0", # For configuration objects
26-
"gymnasium>=0.29.0", # For FrozenLake and other gym environments
27-
"httpx>=0.24.0", # For HTTP client functionality in agent resources
28-
"fireworks-ai>=0.17.19", # For Fireworks Build SDK integration
29-
],
30-
extras_require={
31-
"dev": [
32-
"build",
33-
"twine",
34-
"pytest>=6.0.0",
35-
"pytest-asyncio",
36-
"pytest-httpserver",
37-
"werkzeug>=2.0.0", # needed for test_url_handling.py
38-
"black>=21.5b2",
39-
"mypy>=0.812",
40-
"flake8>=3.9.2",
41-
"autopep8>=1.5.0",
42-
"transformers>=4.0.0",
43-
"types-setuptools",
44-
"types-requests",
45-
"types-PyYAML",
46-
"types-docker",
47-
"versioneer>=0.20",
48-
"openai==1.78.1", # needed for tests using OpenAI types
49-
# datasets, hydra-core, omegaconf moved to core dependencies
50-
"pre-commit",
51-
"e2b>=0.15.0", # Added e2b for E2B environment tests
52-
"docker==7.1.0",
53-
],
54-
"trl": [
55-
"torch>=1.9",
56-
"trl>=0.7.0",
57-
"peft>=0.7.0",
58-
"transformers>=4.0.0",
59-
"accelerate>=0.28.0",
60-
],
61-
"deepseek": [
62-
"difflib>=3.0.0",
63-
],
64-
# "deepeval": ["deepeval>=3.1.0"],
65-
"openevals": ["openevals>=0.1.0"],
66-
},
67-
entry_points={
68-
"console_scripts": [
69-
"fireworks-reward=reward_kit.cli:main",
70-
"reward-kit=reward_kit.cli:main",
71-
"reward-protocol=reward_kit.cli:main",
72-
],
73-
},
74-
author="Fireworks AI",
75-
author_email="info@fireworks.ai",
76-
description="A Python library for defining, testing, and using reward functions",
77-
long_description="A library for defining, testing, and deploying reward functions",
78-
url="https://github.com/fireworks-ai/reward-kit",
79-
classifiers=[
80-
"Programming Language :: Python :: 3",
81-
"License :: OSI Approved :: Apache Software License",
82-
"Operating System :: OS Independent",
83-
],
84-
python_requires=">=3.10",
85-
license_files=["LICENSE"],
867
)

0 commit comments

Comments
 (0)