File size: 1,306 Bytes
e313085
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

[build-system]
requires = ["setuptools>=45", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "openenv-kernel_env"
version = "0.1.0"
description = "Kernel Env environment for OpenEnv"
requires-python = ">=3.10"
dependencies = [
    # Core OpenEnv runtime (provides FastAPI server + HTTP client types)
    # install from github
    # "openenv-core[core] @ git+https://github.com/meta-pytorch/OpenEnv.git",
    "openenv-core[core]>=0.2.1",
    # torch/triton are only needed by the GPU worker; install the `gpu` extra (or use the Docker image)
]

[project.optional-dependencies]
gpu = [
    "torch>=2.5",
    "triton>=3.1",
    "ninja",  # torch.utils.cpp_extension.load_inline (language="cuda")
]
dev = [
    "pytest>=8.0.0",
    "pytest-cov>=4.0.0",
]

[project.scripts]
# Server entry point - enables running via: uv run --project . server
# or: python -m kernel_env.server.app
server = "kernel_env.server.app:main"

[tool.setuptools]
include-package-data = true
packages = ["kernel_env", "kernel_env.server"]
package-dir = { "kernel_env" = ".", "kernel_env.server" = "server" }