Unverified Commit 16f413d9 authored by Aleksana's avatar Aleksana Committed by GitHub
Browse files

python3Packages.ray: update dependencies (#463583)

parents 6196bdde 73d68f90

pkgs/development/python-modules/ray/default.nix

+57 −22
Original line number Diff line number Diff line
@@ -17,7 +17,6 @@ 
  protobuf,

  pyyaml,

  requests,

  watchfiles,



  # optional-dependencies

  # cgraph
@@ -34,28 +33,37 @@ 
  aiohttp-cors,

  colorful,

  opencensus,

  opentelemetry-exporter-prometheus,

  opentelemetry-proto,

  opentelemetry-sdk,

  prometheus-client,

  pydantic,

  py-spy,

  smart-open,

  virtualenv,

  # llm

  async-timeout,

  hf-transfer,

  jsonref,

  ninja,

  # nixl,

  typer,

  vllm,

  # observability

  memray,

  opentelemetry-api,

  opentelemetry-sdk,

  opentelemetry-exporter-otlp,

  # rllib

  dm-tree,

  gymnasium,

  lz4,

  # ormsgpack,

  ormsgpack,

  scipy,

  typer,

  rich,

  # serve

  fastapi,

  starlette,

  uvicorn,

  watchfiles,

  # serve-async-inference

  celery,

  # serve-grpc

  pyopenssl,

  # tune
@@ -125,7 +133,6 @@ buildPythonPackage rec { 
    protobuf

    pyyaml

    requests

    watchfiles

  ];



  optional-dependencies = lib.fix (self: {
@@ -141,6 +148,8 @@ buildPythonPackage rec { 
      ++ self.observability

      ++ self.rllib

      ++ self.serve

      ++ self.serve-async-inference

      ++ self.serve-grpc

      ++ self.train

      ++ self.tune

    );
@@ -160,6 +169,9 @@ buildPythonPackage rec { 
      colorful

      grpcio

      opencensus

      opentelemetry-exporter-prometheus

      opentelemetry-proto

      opentelemetry-sdk

      prometheus-client

      pydantic

      py-spy
@@ -167,22 +179,34 @@ buildPythonPackage rec { 
      smart-open

      virtualenv

    ];

    llm = lib.unique (

      [

        async-timeout

        hf-transfer

        jsonref

        jsonschema

        ninja

        # nixl

        typer

        vllm

      ]

      ++ self.data

      ++ self.serve

    );

    observability = [

      memray

      opentelemetry-api

      opentelemetry-sdk

      opentelemetry-exporter-otlp

    ];

    rllib = [

    rllib = lib.unique (

      [

        dm-tree

        gymnasium

        lz4

      # ormsgpack

        ormsgpack

        pyyaml

        scipy

      typer

      rich

    ];

      ]

      ++ self.tune

    );

    serve = lib.unique (

      [

        fastapi
@@ -193,6 +217,12 @@ buildPythonPackage rec { 
      ]

      ++ self.default

    );

    serve-async-inference = lib.unique (

      [

        celery

      ]

      ++ self.serve

    );

    serve-grpc = lib.unique (

      [

        grpcio
@@ -200,7 +230,12 @@ buildPythonPackage rec { 
      ]

      ++ self.serve

    );

    train = self.tune;

    train = lib.unique (

      [

        pydantic

      ]

      ++ self.tune

    );

    tune = [

      fsspec

      pandas