Gaetan Lepage 2024-11-20 10:08:48 +01:00
parent b413874cd0
commit 4dc27d25a7
2 changed files with 89 additions and 65 deletions

View File

@ -1,11 +1,11 @@
{
cp310 = {
hash = "sha256-On1vcVm85BF7/o+cPQtl/yclf+LdjXN9rQ84aWZkQNo=";
hash = "sha256-Ypj7mBzQ+oYH8ZF96yeSWrit1IxgulvQ9s9A1MxdrOQ=";
};
cp311 = {
hash = "sha256-7OgCzzocECtT9juLyQ2UeXHEs4feryM8Ik7Y7zSh88s=";
hash = "sha256-Ia7hJ64anPYZMAGrQdJVG8yBMxujtxltAA8W0Q8VxwU=";
};
cp312 = {
hash = "sha256-B1B9L5lh6NU5DA62Bt8kkhbvWvsf+BhVgfPpIEHWYpM=";
hash = "sha256-AWkw5rp0uRtAEXpksk97//SKangPI9KwZKej9DvE4aI=";
};
}

View File

@ -8,55 +8,63 @@
autoPatchelfHook,
# dependencies
aiohttp,
aiohttp-cors,
aiosignal,
attrs,
click,
cloudpickle,
colorama,
colorful,
cython,
filelock,
frozenlist,
gpustat,
grpcio,
jsonschema,
msgpack,
numpy,
opencensus,
packaging,
prometheus-client,
psutil,
pydantic,
py-spy,
protobuf,
pyyaml,
requests,
setproctitle,
smart-open,
virtualenv,
watchfiles,
# optional-dependencies
# adag
cupy,
# client
grpcio,
# data
fsspec,
numpy,
pandas,
pyarrow,
# default
aiohttp,
aiohttp-cors,
colorful,
opencensus,
prometheus-client,
pydantic,
py-spy,
smart-open,
virtualenv,
# observability
opentelemetry-api,
opentelemetry-sdk,
opentelemetry-exporter-otlp,
# rllib
dm-tree,
gym,
gymnasium,
lz4,
matplotlib,
scikit-image,
scipy,
aiorwlock,
typer,
rich,
# serve
fastapi,
starlette,
uvicorn,
tabulate,
# serve-grpc
pyopenssl,
# tune
tensorboardx,
}:
let
pname = "ray";
version = "2.38.0";
version = "2.39.0";
in
buildPythonPackage rec {
inherit pname version;
@ -84,68 +92,84 @@ buildPythonPackage rec {
autoPatchelfHook
];
pythonRelaxDeps = [
"click"
"grpcio"
"protobuf"
"virtualenv"
];
dependencies = [
aiohttp
aiohttp-cors
aiosignal
attrs
click
cloudpickle
colorama
colorful
cython
aiosignal
filelock
frozenlist
gpustat
grpcio
jsonschema
msgpack
numpy
opencensus
packaging
prometheus-client
psutil
pydantic
py-spy
protobuf
pyyaml
requests
setproctitle
smart-open
virtualenv
watchfiles
];
optional-dependencies = rec {
air-deps = data-deps ++ serve-deps ++ tune-deps ++ rllib-deps;
data-deps = [
adag = [
cupy
];
air = lib.unique (data ++ serve ++ tune ++ train);
all = lib.flatten (builtins.attrValues optional-dependencies);
client = [ grpcio ];
data = [
fsspec
numpy
pandas
pyarrow
];
rllib-deps = tune-deps ++ [
default = [
aiohttp
aiohttp-cors
colorful
grpcio
opencensus
prometheus-client
pydantic
py-spy
requests
smart-open
virtualenv
];
observability = [
opentelemetry-api
opentelemetry-sdk
opentelemetry-exporter-otlp
];
rllib = [
dm-tree
gym
gymnasium
lz4
matplotlib
pyyaml
scikit-image
scipy
typer
rich
];
serve-deps = [
aiorwlock
fastapi
serve = lib.unique (
[
fastapi
requests
starlette
uvicorn
watchfiles
]
++ default
);
serve-grpc = lib.unique (
[
grpcio
pyopenssl
]
++ serve
);
train = tune;
tune = [
fsspec
pandas
starlette
uvicorn
];
tune-deps = [
tabulate
pyarrow
requests
tensorboardx
];
};