# Maintainer: acxz <akashpatel2008 at yahoo dot com>

pkgname=python-ray
pkgver=1.11.0
pkgrel=1
pkgdesc='A fast and simple framework for building and running distributed
applications.'
arch=('x86_64')
url='https://ray.io'
license=('Apache-2.0')
depends=(psmisc python python-attrs python-click python-filelock python-grpcio
         python-jsonschema python-msgpack python-numpy python-protobuf
         python-pyaml python-redis)
optdepends=(
            'python-pandas: for ray[data]'
            'python-pyarrow: for ray[data]'
            'python-fsspec: for ray[data]'
            'python-aiohttp: for ray[default, serve]'
            'python-aiosignal: for ray[default, serve]'
            'python-aiohttps_cors: for ray[default, serve]'
            'python-aioredis: for ray[default, serve]'
            'python-colorful: for ray[default, serve]'
            'python-frozenlist: for ray[default, serve]'
            'py-spy: for ray[default, serve]'
            'python-requests: for ray[default, serve, tune]'
            'python-gpustat: for ray[default, serve]'
            'python-opencensus: for ray[default, serve]'
            'python-prometheus-client: for ray[default, serve]'
            'python-smart-open: for ray[default, serve]'
            'uvicorn: for ray[serve]'
            'python-starlette: for ray[serve]'
            'python-fastapi: for ray[serve]'
            'python-pandas: for ray[tune, rllib]'
            'python-tabulate: for ray[tune, rllib]'
            'python-tensorboardx: for ray[tune, rllib]'
            'python-kubernetes: for ray[k8s]'
            'python-urllib3: for ray[k8s]'
            'python-kopf: for ray[k8s]'
            'python-opentelemetry-api: for ray[observability]'
            'python-opentelemetry-sdk: for ray[observability]'
            'python-opentelemetry-exporter-otlp: for ray[observability]'
            'python-dm-tree: for ray[rllib]'
            'python-gym: for ray[rllib]'
            'python-lz4: for ray[rllib]'
            'python-scikit-image: for ray[rllib]'
            'python-pyaml: for ray[rllib]'
            'python-scipy: for ray[rllib]'
           )
makedepends=(python python-setuptools python-wheel python-pip cython bazel)
_pkgname=ray
source=("${pkgname}-${pkgver}.tar.gz::https://github.com/ray-project/ray/archive/${_pkgname}-$pkgver.tar.gz"
        "grpc.patch"::"https://patch-diff.githubusercontent.com/raw/ray-project/ray/pull/21866.patch"
        "py310.patch"::"https://patch-diff.githubusercontent.com/raw/ray-project/ray/pull/21221.patch")
sha256sums=('efc36ce277bc650eaeb4cec959f5280b348b0a25347e30dfd9e57fa6e5e0760b'
            'SKIP'
            'SKIP')

prepare() {
  cd "${srcdir}/${_pkgname}-${_pkgname}-${pkgver}"

  patch --strip=1 < "${srcdir}/grpc.patch"
  #patch --strip=1 < "${srcdir}/py310.patch"
  sed -i "s/9)]/9), (3, 10)]/" python/setup.py

  # https://aur.archlinux.org/packages/py-spy#comment-781521
  sed -i "/py-spy/d" python/setup.py
}

build() {
  cd "${srcdir}/${_pkgname}-${_pkgname}-${pkgver}/python"
  python setup.py build
}

package() {
  cd "${srcdir}/${_pkgname}-${_pkgname}-${pkgver}/python"
  python setup.py install --root="$pkgdir"/ --optimize=1
}