diff --git a/pkgs/development/python-modules/energyflow/default.nix b/pkgs/development/python-modules/energyflow/default.nix index cf2850757686..fffc8b1998ed 100644 --- a/pkgs/development/python-modules/energyflow/default.nix +++ b/pkgs/development/python-modules/energyflow/default.nix @@ -1,19 +1,27 @@ { lib, + stdenv, buildPythonPackage, fetchFromGitHub, - h5py, + + # build-system hatch-vcs, hatchling, - igraph, + + # dependencies + h5py, numpy, - pot, - pytestCheckHook, - pythonOlder, + wasserstein, + + # optional-dependencies + igraph, scikit-learn, tensorflow, + + # tests + pot, + pytestCheckHook, tf-keras, - wasserstein, }: buildPythonPackage rec { @@ -21,8 +29,6 @@ buildPythonPackage rec { version = "1.4.0"; pyproject = true; - disabled = pythonOlder "3.9"; - src = fetchFromGitHub { owner = "pkomiske"; repo = "EnergyFlow"; @@ -60,21 +66,28 @@ buildPythonPackage rec { tf-keras ] ++ lib.flatten (builtins.attrValues optional-dependencies); - disabledTests = [ - # Issues with array - "test_emd_equivalence" - "test_gdim" - "test_n_jobs" - "test_periodic_phi" - ]; + disabledTests = + [ + # Issues with array + "test_emd_equivalence" + "test_gdim" + "test_n_jobs" + "test_periodic_phi" + ] + ++ lib.optionals stdenv.hostPlatform.isDarwin [ + # RuntimeError: EMDStatus - Infeasible + "test_emd_byhand_1_1" + "test_emd_return_flow" + "test_emde" + ]; pythonImportsCheck = [ "energyflow" ]; - meta = with lib; { + meta = { description = "Python package for the EnergyFlow suite of tools"; homepage = "https://energyflow.network/"; changelog = "https://github.com/thaler-lab/EnergyFlow/releases/tag/v${version}"; - license = licenses.gpl3Only; - maintainers = with maintainers; [ veprbl ]; + license = lib.licenses.gpl3Only; + maintainers = with lib.maintainers; [ veprbl ]; }; }