nixpkgs/pkgs/development/python-modules/iopath/default.nix
2023-07-10 15:22:16 +08:00

69 lines
1.3 KiB
Nix

{ lib
, stdenv
, buildPythonPackage
, fetchFromGitHub
, pythonOlder
# build inputs
, tqdm
, portalocker
, boto3
# check inputs
, pytestCheckHook
, torch
}:
let
pname = "iopath";
version = "0.1.9";
in
buildPythonPackage {
inherit pname version;
format = "setuptools";
disabled = pythonOlder "3.10";
src = fetchFromGitHub {
owner = "facebookresearch";
repo = "iopath";
rev = "refs/tags/v${version}";
hash = "sha256-Qubf/mWKMgYz9IVoptMZrwy4lQKsNGgdqpJB1j/u5s8=";
};
propagatedBuildInputs = [
tqdm
portalocker
];
nativeCheckInputs = [
pytestCheckHook
torch
];
disabledTests = [
# requires network access
"test_download"
"test_bad_args"
];
disabledTestPaths = [
# flakey
"tests/async_torch_test.py"
"tests/async_writes_test.py"
];
pythonImportsCheck = [
"iopath"
];
passthru.optional-dependencies = {
aws = [ boto3 ];
};
meta = with lib; {
description = "A python library that provides common I/O interface across different storage backends.";
homepage = "https://github.com/facebookresearch/iopath";
changelog = "https://github.com/facebookresearch/iopath/releases/tag/v${version}";
license = licenses.mit;
maintainers = with maintainers; [ happysalada ];
};
}