Unverified Commit 7f88038d authored by Sandro Jäckel's avatar Sandro Jäckel Committed by GitHub
Browse files

python312Packages.torch-geometric: init at 2.6.1,... 

python312Packages.torch-geometric: init at 2.6.1, python312Packages.compressai: 1.2.4 -> 1.2.6 (#345934)
parents 8ab962f8 93d720f0

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

+21 −12
Original line number Diff line number Diff line 
{

  lib,

  buildPythonPackage,

  pythonOlder,

  fetchFromGitHub,



  # build-system

  pybind11,

  setuptools,

  wheel,



  # dependencies

  einops,

  numpy,

  matplotlib,

  pandas,

  pytorch-msssim,

  scipy,

  torch,

  torch-geometric,

  torchvision,



  # optional-dependencies

  ipywidgets,

  jupyter,



  # tests

  plotly,

  pytestCheckHook,

}:



buildPythonPackage rec {

  pname = "compressai";

  version = "1.2.4";

  version = "1.2.6";

  pyproject = true;



  disabled = pythonOlder "3.6";



  src = fetchFromGitHub {

    owner = "InterDigitalInc";

    repo = "CompressAI";

    rev = "refs/tags/v${version}";

    hash = "sha256-nT2vd7t67agIWobJalORbRuns0UJGRGGbTX2/8vbTiY=";

    hash = "sha256-xvzhhLn0iBzq3h1nro8/83QWEQe9K4zRa3RSZk+hy3Y=";

    fetchSubmodules = true;

  };



  nativeBuildInputs = [

  build-system = [

    pybind11

    setuptools

    wheel

  ];



  propagatedBuildInputs = [

  dependencies = [

    einops

    numpy

    matplotlib

    pandas

    pytorch-msssim

    scipy

    torch

    torch-geometric

    torchvision

  ];
@@ -81,10 +90,10 @@ buildPythonPackage rec { 
    "test_pretrained"

  ];



  meta = with lib; {

  meta = {

    description = "PyTorch library and evaluation platform for end-to-end compression research";

    homepage = "https://github.com/InterDigitalInc/CompressAI";

    license = licenses.bsd3Clear;

    maintainers = with maintainers; [ GaetanLepage ];

    license = lib.licenses.bsd3Clear;

    maintainers = with lib.maintainers; [ GaetanLepage ];

  };

}

pkgs/development/python-modules/torch-geometric/default.nix

0 → 100644
+179 −0
Original line number Diff line number Diff line 
{

  lib,

  buildPythonPackage,

  fetchFromGitHub,



  # build-system

  flit-core,



  # dependencies

  aiohttp,

  fsspec,

  jinja2,

  numpy,

  psutil,

  pyparsing,

  requests,

  torch,

  tqdm,



  # optional-dependencies

  matplotlib,

  networkx,

  pandas,

  protobuf,

  wandb,

  ipython,

  matplotlib-inline,

  pre-commit,

  torch-geometric,

  ase,

  # captum,

  graphviz,

  h5py,

  numba,

  opt-einsum,

  pgmpy,

  pynndescent,

  # pytorch-memlab,

  rdflib,

  rdkit,

  scikit-image,

  scikit-learn,

  scipy,

  statsmodels,

  sympy,

  tabulate,

  torchmetrics,

  trimesh,

  pytorch-lightning,

  yacs,

  huggingface-hub,

  onnx,

  onnxruntime,

  pytest,

  pytest-cov,



  # tests

  pytestCheckHook,

}:



buildPythonPackage rec {

  pname = "torch-geometric";

  version = "2.6.1";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "pyg-team";

    repo = "pytorch_geometric";

    rev = "refs/tags/${version}";

    hash = "sha256-Zw9YqPQw2N0ZKn5i5Kl4Cjk9JDTmvZmyO/VvIVr6fTU=";

  };



  build-system = [

    flit-core

  ];



  dependencies = [

    aiohttp

    fsspec

    jinja2

    numpy

    psutil

    pyparsing

    requests

    torch

    tqdm

  ];



  optional-dependencies = {

    benchmark = [

      matplotlib

      networkx

      pandas

      protobuf

      wandb

    ];

    dev = [

      ipython

      matplotlib-inline

      pre-commit

      torch-geometric

    ];

    full = [

      ase

      # captum

      graphviz

      h5py

      matplotlib

      networkx

      numba

      opt-einsum

      pandas

      pgmpy

      pynndescent

      # pytorch-memlab

      rdflib

      rdkit

      scikit-image

      scikit-learn

      scipy

      statsmodels

      sympy

      tabulate

      torch-geometric

      torchmetrics

      trimesh

    ];

    graphgym = [

      protobuf

      pytorch-lightning

      yacs

    ];

    modelhub = [

      huggingface-hub

    ];

    test = [

      onnx

      onnxruntime

      pytest

      pytest-cov

    ];

  };



  pythonImportsCheck = [

    "torch_geometric"

  ];



  nativeCheckInputs = [

    pytestCheckHook

  ];



  preCheck = ''

    export HOME=$(mktemp -d)

  '';



  disabledTests = [

    # TODO: try to re-enable when triton will have been updated to 3.0

    # torch._dynamo.exc.BackendCompilerFailed: backend='inductor' raised:

    # LoweringException: ImportError: cannot import name 'triton_key' from 'triton.compiler.compiler'

    "test_compile_hetero_conv_graph_breaks"

    "test_compile_multi_aggr_sage_conv"



    # RuntimeError: addmm: computation on CPU is not implemented for SparseCsr + SparseCsr @ SparseCsr without MKL.

    # PyTorch built with MKL has better support for addmm with sparse CPU tensors.

    "test_asap"

    "test_graph_unet"



    # AttributeError: type object 'Any' has no attribute '_name'

    "test_type_repr"

  ];



  meta = {

    description = "Graph Neural Network Library for PyTorch";

    homepage = "https://github.com/pyg-team/pytorch_geometric";

    changelog = "https://github.com/pyg-team/pytorch_geometric/blob/${src.rev}/CHANGELOG.md";

    license = lib.licenses.mit;

    maintainers = with lib.maintainers; [ GaetanLepage ];

  };

}

pkgs/top-level/python-packages.nix

+2 −0
Original line number Diff line number Diff line
@@ -15692,6 +15692,8 @@ self: super: with self; { 














  torch-audiomentations = callPackage ../development/python-modules/torch-audiomentations { };



























  torch-geometric = callPackage ../development/python-modules/torch-geometric { };



























  torch-pitch-shift = callPackage ../development/python-modules/torch-pitch-shift { };





























  torch-bin = callPackage ../development/python-modules/torch/bin.nix {