69 lines
1.6 KiB
Nix
69 lines
1.6 KiB
Nix
{ lib
|
|
, buildPythonPackage
|
|
, fetchFromGitHub
|
|
, torch
|
|
, ninja
|
|
, scipy
|
|
, which
|
|
, pybind11
|
|
, pytest-xdist
|
|
, pytestCheckHook
|
|
}:
|
|
|
|
let
|
|
linePatch = ''
|
|
import os
|
|
os.environ['PATH'] = os.environ['PATH'] + ':${ninja}/bin'
|
|
'';
|
|
in
|
|
buildPythonPackage rec {
|
|
pname = "deepwave";
|
|
version = "0.0.17";
|
|
format = "pyproject";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "ar4";
|
|
repo = pname;
|
|
rev = "v${version}";
|
|
sha256 = "sha256-4B3V87/voYs61pXhqmydLe48JsnRGuJlUYOOdmJlroA=";
|
|
};
|
|
|
|
# unable to find ninja although it is available, most likely because it looks for its pip version
|
|
postPatch = ''
|
|
substituteInPlace setup.cfg --replace "ninja" ""
|
|
|
|
# Adding ninja to the path forcibly
|
|
mv src/deepwave/__init__.py tmp
|
|
echo "${linePatch}" > src/deepwave/__init__.py
|
|
cat tmp >> src/deepwave/__init__.py
|
|
rm tmp
|
|
'';
|
|
|
|
# The source files are compiled at runtime and cached at the
|
|
# $HOME/.cache folder, so for the check phase it is needed to
|
|
# have a temporary home. This is also the reason ninja is not
|
|
# needed at the nativeBuildInputs, since it will only be used
|
|
# at runtime.
|
|
preBuild = ''
|
|
export HOME=$(mktemp -d)
|
|
'';
|
|
|
|
propagatedBuildInputs = [ torch pybind11 ];
|
|
|
|
checkInputs = [
|
|
which
|
|
scipy
|
|
pytest-xdist
|
|
pytestCheckHook
|
|
];
|
|
|
|
pythonImportsCheck = [ "deepwave" ];
|
|
|
|
meta = with lib; {
|
|
description = "Wave propagation modules for PyTorch";
|
|
homepage = "https://github.com/ar4/deepwave";
|
|
license = licenses.mit;
|
|
platforms = intersectLists platforms.x86_64 platforms.linux;
|
|
maintainers = with maintainers; [ atila ];
|
|
};
|
|
}
|