python3Packages.torchdata: init at 0.11.0 (#460931)

{

  lib,

  buildPythonPackage,

  fetchFromGitHub,

  # build-system

  cmake,

  ninja,

  setuptools,

  torch,

  # dependencies

  requests,

  urllib3,

  # tests

  datasets,

  parameterized,

  pytest-xdist,

  pytestCheckHook,

}:

buildPythonPackage rec {

  pname = "torchdata";

  version = "0.11.0";

  pyproject = true;

  src = fetchFromGitHub {

    owner = "meta-pytorch";

    repo = "data";

    tag = "v${version}";

    hash = "sha256-TSkZLL4WDSacuX4tl0+1bKSJCRI3LEhAyU3ztdlUvgk=";

  };

  build-system = [

    cmake

    ninja

    setuptools

    torch

  ];

  dontUseCmakeConfigure = true;

  dependencies = [

    requests

    urllib3

  ];

  pythonImportsCheck = [ "torchdata" ];

  nativeCheckInputs = [

    datasets

    parameterized

    pytest-xdist

    pytestCheckHook

  ];

  meta = {

    description = "Iterative enhancement to the PyTorch torch.utils.data.DataLoader and torch.utils.data.Dataset/IterableDataset";

    homepage = "https://github.com/meta-pytorch/data";

    changelog = "https://github.com/meta-pytorch/data/releases/tag/${src.tag}";

    license = lib.licenses.bsd3;

    maintainers = with lib.maintainers; [ GaetanLepage ];

  };

}

  torchcrepe = callPackage ../development/python-modules/torchcrepe { };

  torchdata = callPackage ../development/python-modules/torchdata { };

  torchdiffeq = callPackage ../development/python-modules/torchdiffeq { };

  torcheval = callPackage ../development/python-modules/torcheval { };