jtowarek commited on
Commit
47250b3
·
verified ·
1 Parent(s): ed4bdac

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. __init__.py +10 -0
  2. openenv.yaml +3 -2
  3. pyproject.toml +34 -26
__init__.py CHANGED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ """KantBench Environment — 90+ game theory games for LLM training."""
2
+
3
+ from .client import KantBenchEnv
4
+ from .models import KantBenchAction, KantBenchObservation
5
+
6
+ __all__ = [
7
+ "KantBenchAction",
8
+ "KantBenchObservation",
9
+ "KantBenchEnv",
10
+ ]
openenv.yaml CHANGED
@@ -1,6 +1,7 @@
1
  spec_version: 1
2
- name: kant
3
  type: space
4
  runtime: fastapi
5
- app: env.app:app
6
  port: 8000
 
 
1
  spec_version: 1
2
+ name: KantBench
3
  type: space
4
  runtime: fastapi
5
+ app: server.app:app
6
  port: 8000
7
+
pyproject.toml CHANGED
@@ -1,37 +1,45 @@
 
 
 
 
 
 
1
  [build-system]
2
- requires = ["setuptools>=68.0", "wheel"]
3
- build-backend = "setuptools.backends._legacy:_Backend"
4
 
5
  [project]
6
- name = "openenv-kant"
7
  version = "0.1.0"
8
- description = "Kant: Game-theory benchmark for AI agents on OpenEnv"
9
  requires-python = ">=3.10"
10
- license = { text = "BSD-3-Clause" }
11
  dependencies = [
12
- "openenv-core[core]>=0.2.1",
13
- "fastapi>=0.115.0",
14
- "pydantic>=2.0.0",
15
- "uvicorn>=0.24.0",
16
- "numpy>=1.24.0",
 
 
 
 
 
 
 
17
  ]
18
 
19
  [project.optional-dependencies]
20
- dev = ["pytest>=7.0", "pytest-cov>=4.0", "pytest-asyncio>=0.21.0"]
21
- gradio = ["gradio>=4.0.0", "matplotlib>=3.7.0"]
22
- train = ["transformers>=4.40.0", "trl>=0.12.0", "peft>=0.13.0", "datasets>=2.20.0", "accelerate>=0.34.0", "bitsandbytes>=0.43.0", "wandb>=0.18.0"]
23
- api = ["openai>=1.0.0", "anthropic>=0.30.0"]
24
- eval = ["lm-eval>=0.4.0", "datasets>=2.20.0"]
25
-
26
- [tool.pytest.ini_options]
27
- testpaths = ["tests"]
28
- asyncio_mode = "auto"
29
-
30
- [tool.ruff]
31
- line-length = 88
32
 
33
- [tool.ruff.lint]
34
- select = ["E", "F", "I"]
 
 
35
 
36
- [tool.setuptools.packages.find]
37
- where = ["."]
 
 
 
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+ # All rights reserved.
3
+ #
4
+ # This source code is licensed under the BSD-style license found in the
5
+ # LICENSE file in the root directory of this source tree.
6
+
7
  [build-system]
8
+ requires = ["setuptools>=45", "wheel"]
9
+ build-backend = "setuptools.build_meta"
10
 
11
  [project]
12
+ name = "openenv-KantBench"
13
  version = "0.1.0"
14
+ description = "Kantbench environment for OpenEnv"
15
  requires-python = ">=3.10"
 
16
  dependencies = [
17
+ # Core OpenEnv runtime (provides FastAPI server + HTTP client types)
18
+ # install from github
19
+ # "openenv-core[core] @ git+https://github.com/meta-pytorch/OpenEnv.git",
20
+ "openenv-core[core]==0.2.1",
21
+ # Environment-specific dependencies
22
+ # Add all dependencies needed for your environment here
23
+ # Examples:
24
+ # "numpy>=1.19.0",
25
+ # "torch>=2.0.0",
26
+ # "gymnasium>=0.29.0",
27
+ # "openspiel>=1.0.0",
28
+ # "smolagents>=1.22.0,<2",
29
  ]
30
 
31
  [project.optional-dependencies]
32
+ dev = [
33
+ "pytest>=8.0.0",
34
+ "pytest-cov>=4.0.0",
35
+ ]
 
 
 
 
 
 
 
 
36
 
37
+ [project.scripts]
38
+ # Server entry point - enables running via: uv run --project . server
39
+ # or: python -m KantBench.server.app
40
+ server = "KantBench.server.app:main"
41
 
42
+ [tool.setuptools]
43
+ include-package-data = true
44
+ packages = ["KantBench", "KantBench.server"]
45
+ package-dir = { "KantBench" = ".", "KantBench.server" = "server" }