diff options
Diffstat (limited to 'pkgs/development/python-modules/ray/default.nix')
-rw-r--r-- | pkgs/development/python-modules/ray/default.nix | 116 |
1 files changed, 57 insertions, 59 deletions
diff --git a/pkgs/development/python-modules/ray/default.nix b/pkgs/development/python-modules/ray/default.nix index d0869e8afc080..ba2d8afe9be97 100644 --- a/pkgs/development/python-modules/ray/default.nix +++ b/pkgs/development/python-modules/ray/default.nix @@ -1,59 +1,62 @@ { + lib, + buildPythonPackage, + pythonOlder, + pythonAtLeast, + python, + fetchPypi, + autoPatchelfHook, + + # dependencies aiohttp, aiohttp-cors, - aiorwlock, aiosignal, attrs, - autoPatchelfHook, - buildPythonPackage, - fetchPypi, click, cloudpickle, colorama, colorful, cython, - dm-tree, - fastapi, filelock, frozenlist, - fsspec, gpustat, grpcio, - gym, jsonschema, - lib, - lz4, - matplotlib, msgpack, numpy, opencensus, packaging, - pandas, - py-spy, prometheus-client, psutil, - pyarrow, pydantic, - python, - pythonAtLeast, - pythonOlder, - pythonRelaxDepsHook, + py-spy, pyyaml, requests, - scikit-image, - scipy, setproctitle, smart-open, + virtualenv, + + # optional-dependencies + fsspec, + pandas, + pyarrow, + dm-tree, + gym, + lz4, + matplotlib, + scikit-image, + scipy, + aiorwlock, + fastapi, starlette, + uvicorn, tabulate, tensorboardx, - uvicorn, - virtualenv, }: let pname = "ray"; - version = "2.31.0"; + version = "2.34.0"; in buildPythonPackage rec { inherit pname version; @@ -77,42 +80,8 @@ buildPythonPackage rec { // binary-hash ); - passthru.optional-dependencies = rec { - data-deps = [ - pandas - pyarrow - fsspec - ]; - - serve-deps = [ - aiorwlock - fastapi - pandas - starlette - uvicorn - ]; - - tune-deps = [ - tabulate - tensorboardx - ]; - - rllib-deps = tune-deps ++ [ - dm-tree - gym - lz4 - matplotlib - scikit-image - pyyaml - scipy - ]; - - air-deps = data-deps ++ serve-deps ++ tune-deps ++ rllib-deps; - }; - nativeBuildInputs = [ autoPatchelfHook - pythonRelaxDepsHook ]; pythonRelaxDeps = [ @@ -123,10 +92,10 @@ buildPythonPackage rec { ]; dependencies = [ - attrs aiohttp aiohttp-cors aiosignal + attrs click cloudpickle colorama @@ -141,10 +110,10 @@ buildPythonPackage rec { numpy opencensus packaging - py-spy prometheus-client psutil pydantic + py-spy pyyaml requests setproctitle @@ -152,6 +121,35 @@ buildPythonPackage rec { virtualenv ]; + optional-dependencies = rec { + air-deps = data-deps ++ serve-deps ++ tune-deps ++ rllib-deps; + data-deps = [ + fsspec + pandas + pyarrow + ]; + rllib-deps = tune-deps ++ [ + dm-tree + gym + lz4 + matplotlib + pyyaml + scikit-image + scipy + ]; + serve-deps = [ + aiorwlock + fastapi + pandas + starlette + uvicorn + ]; + tune-deps = [ + tabulate + tensorboardx + ]; + }; + postInstall = '' chmod +x $out/${python.sitePackages}/ray/core/src/ray/{gcs/gcs_server,raylet/raylet} ''; |