36 lines
1.5 KiB
Python
36 lines
1.5 KiB
Python
# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other
|
|
# Spack Project Developers. See the top-level COPYRIGHT file for details.
|
|
#
|
|
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
|
|
|
|
from spack.package import *
|
|
|
|
|
|
class PyOgb(PythonPackage):
|
|
"""The Open Graph Benchmark (OGB) is a collection of benchmark datasets, data
|
|
loaders, and evaluators for graph machine learning. Datasets cover a variety of
|
|
graph machine learning tasks and real-world applications. The OGB data loaders
|
|
are fully compatible with popular graph deep learning frameworks, including
|
|
PyTorch Geometric and Deep Graph Library (DGL). They provide automatic dataset
|
|
downloading, standardized dataset splits, and unified performance evaluation.
|
|
"""
|
|
|
|
homepage = "https://github.com/snap-stanford/ogb"
|
|
pypi = "ogb/ogb-1.3.5.tar.gz"
|
|
|
|
maintainers("meyersbs")
|
|
|
|
license("MIT")
|
|
|
|
version("1.3.5", sha256="ac958094ac3019822e742155b82cb2bf02830aa72a4264ba9ee09b288f0c080c")
|
|
|
|
depends_on("py-setuptools", type="build")
|
|
depends_on("py-torch@1.6.0:", type=("build", "run"))
|
|
depends_on("py-numpy@1.16.0:", type=("build", "run"))
|
|
depends_on("py-tqdm@4.29.0:", type=("build", "run"))
|
|
depends_on("py-scikit-learn@0.20.0:", type=("build", "run"))
|
|
depends_on("py-pandas@0.24.0:", type=("build", "run"))
|
|
depends_on("py-six@1.12.0:", type=("build", "run"))
|
|
depends_on("py-urllib3@1.24.0:", type=("build", "run"))
|
|
depends_on("py-outdated@0.2.0:", type=("build", "run"))
|