meta.yaml

# Copyright (c) 2022-2024, NVIDIA CORPORATION.

{% set version = environ['RAPIDS_PACKAGE_VERSION'].lstrip('v') + environ.get('VERSION_SUFFIX', '') %}
{% set minor_version = version.split('.')[0] + '.' + version.split('.')[1] %}
{% set py_version = environ['CONDA_PY'] %}
{% set date_string = environ['RAPIDS_DATE_STRING'] %}

package:
  name: cugraph-pyg
  version: {{ version }}

source:
  path: ../../..

build:
  number: {{ GIT_DESCRIBE_NUMBER }}
  string: py{{ py_version }}_{{ date_string }}_{{ GIT_DESCRIBE_HASH }}_{{ GIT_DESCRIBE_NUMBER }}
  script_env:
    - PARALLEL_LEVEL

requirements:
  build:
    - {{ stdlib("c") }}
  host:
    - cython >=3.0.0
    - python
    - rapids-build-backend>=0.3.1,<0.4.0.dev0
    - setuptools>=61.0.0
  run:
    - rapids-dask-dependency ={{ minor_version }}
    - numba >=0.57
    - numpy >=1.23,<3.0a0
    - pandas
    - python
    - pytorch >=2.3,<2.4.0a0
    - cupy >=12.0.0
    - cugraph ={{ minor_version }}
    - pylibcugraphops ={{ minor_version }}
    - tensordict >=0.1.2
    - pytorch_geometric >=2.5,<2.6

tests:
  imports:
    - cugraph_pyg

about:
  home: https://rapids.ai/
  dev_url: https://github.com/rapidsai/cugraph
  license: Apache-2.0
  license_file: ../../../LICENSE
  summary: cuGraph-pyg library