88 lines
1.9 KiB
Nix
88 lines
1.9 KiB
Nix
{ lib
|
|
, pythonOlder
|
|
, pythonAtLeast
|
|
, buildPythonPackage
|
|
, fetchFromGitHub
|
|
, fetchpatch
|
|
# Python Inputs
|
|
, fastdtw
|
|
, numpy
|
|
, psutil
|
|
, qiskit-terra
|
|
, scikit-learn
|
|
, sparse
|
|
# Optional inputs
|
|
, withTorch ? true
|
|
, pytorch
|
|
# Check Inputs
|
|
, pytestCheckHook
|
|
, ddt
|
|
, pytest-timeout
|
|
, qiskit-aer
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "qiskit-machine-learning";
|
|
version = "0.4.0";
|
|
|
|
disabled = pythonOlder "3.6";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "qiskit";
|
|
repo = pname;
|
|
rev = "refs/tags/${version}";
|
|
sha256 = "sha256-WZSXt+sVeO64wCVbDgBhuGvo5jTn/JKh9oNSO7ZY9wo=";
|
|
};
|
|
|
|
propagatedBuildInputs = [
|
|
fastdtw
|
|
numpy
|
|
psutil
|
|
qiskit-terra
|
|
scikit-learn
|
|
sparse
|
|
] ++ lib.optional withTorch pytorch;
|
|
|
|
doCheck = false; # TODO: enable. Tests fail on unstable due to some multithreading issue?
|
|
checkInputs = [
|
|
pytestCheckHook
|
|
pytest-timeout
|
|
ddt
|
|
qiskit-aer
|
|
];
|
|
|
|
pythonImportsCheck = [ "qiskit_machine_learning" ];
|
|
|
|
pytestFlagsArray = [
|
|
"--durations=10"
|
|
"--showlocals"
|
|
"-vv"
|
|
"--ignore=test/connectors/test_torch_connector.py" # TODO: fix, get multithreading errors with python3.9, segfaults
|
|
];
|
|
disabledTests = [
|
|
# Slow tests >10 s
|
|
"test_readme_sample"
|
|
"test_vqr_8"
|
|
"test_vqr_7"
|
|
"test_qgan_training_cg"
|
|
"test_vqc_4"
|
|
"test_classifier_with_circuit_qnn_and_cross_entropy_4"
|
|
"test_vqr_4"
|
|
"test_regressor_with_opflow_qnn_4"
|
|
"test_qgan_save_model"
|
|
"test_qgan_training_analytic_gradients"
|
|
"test_qgan_training_run_algo_numpy"
|
|
"test_ad_hoc_data"
|
|
"test_qgan_training"
|
|
];
|
|
|
|
meta = with lib; {
|
|
description = "Software for developing quantum computing programs";
|
|
homepage = "https://qiskit.org";
|
|
downloadPage = "https://github.com/QISKit/qiskit-optimization/releases";
|
|
changelog = "https://qiskit.org/documentation/release_notes.html";
|
|
license = licenses.asl20;
|
|
maintainers = with maintainers; [ drewrisinger ];
|
|
};
|
|
}
|