123 lines
3.2 KiB
Nix
123 lines
3.2 KiB
Nix
{ stdenv
|
|
, lib
|
|
, buildPythonPackage
|
|
, pythonOlder
|
|
, fetchFromGitHub
|
|
, fetchpatch
|
|
, freezegun
|
|
, google-api-core
|
|
, matplotlib
|
|
, networkx
|
|
, numpy
|
|
, pandas
|
|
, protobuf
|
|
, requests
|
|
, scipy
|
|
, sortedcontainers
|
|
, sympy
|
|
, typing-extensions
|
|
# test inputs
|
|
, pytestCheckHook
|
|
, pytest-asyncio
|
|
, pytest-benchmark
|
|
, ply
|
|
, pydot
|
|
, pyyaml
|
|
, pygraphviz
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "cirq";
|
|
version = "0.9.1";
|
|
|
|
disabled = pythonOlder "3.6";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "quantumlib";
|
|
repo = "cirq";
|
|
rev = "v${version}";
|
|
sha256 = "0mygvpq7kzga8l1w2jvwv9a2n3akpss45hrx250gdrnqjp6xrw64";
|
|
};
|
|
|
|
postPatch = ''
|
|
substituteInPlace requirements.txt \
|
|
--replace "freezegun~=0.3.15" "freezegun" \
|
|
--replace "matplotlib~=3.0" "matplotlib" \
|
|
--replace "networkx~=2.4" "networkx" \
|
|
--replace "numpy~=1.16" "numpy" \
|
|
--replace "protobuf~=3.12.0" "protobuf"
|
|
|
|
# Fix serialize_sympy_constants test by allowing small errors in pi
|
|
substituteInPlace cirq/google/arg_func_langs_test.py \
|
|
--replace "'float_value': float(str(np.float32(sympy.pi)))" "'float_value': pytest.approx(float(str(np.float32(sympy.pi))))"
|
|
'';
|
|
|
|
propagatedBuildInputs = [
|
|
freezegun
|
|
google-api-core
|
|
numpy
|
|
matplotlib
|
|
networkx
|
|
pandas
|
|
protobuf
|
|
requests
|
|
scipy
|
|
sortedcontainers
|
|
sympy
|
|
typing-extensions
|
|
];
|
|
|
|
doCheck = true;
|
|
# pythonImportsCheck = [ "cirq" "cirq.Circuit" ]; # cirq's importlib hook doesn't work here
|
|
dontUseSetuptoolsCheck = true;
|
|
checkInputs = [
|
|
pytestCheckHook
|
|
pytest-asyncio
|
|
pytest-benchmark
|
|
ply
|
|
pydot
|
|
pyyaml
|
|
pygraphviz
|
|
];
|
|
|
|
pytestFlagsArray = [
|
|
"--ignore=dev_tools" # Only needed when developing new code, which is out-of-scope
|
|
"--ignore=cirq/contrib/" # requires external (unpackaged) python packages, so untested.
|
|
"--benchmark-disable" # Don't need to run benchmarks when packaging.
|
|
];
|
|
disabledTests = [
|
|
"test_convert_to_ion_gates" # fails on some systems due to rounding error, 0.75 != 0.750...2
|
|
] ++ lib.optionals stdenv.isAarch64 [
|
|
# Seem to fail due to math issues on aarch64?
|
|
"expectation_from_wavefunction"
|
|
"test_single_qubit_op_to_framed_phase_form_output_on_example_case"
|
|
] ++ [
|
|
# slow tests, for quicker building
|
|
"test_anneal_search_method_calls"
|
|
"test_density_matrix_from_state_tomography_is_correct"
|
|
"test_example_runs_qubit_characterizations"
|
|
"test_example_runs_hello_line_perf"
|
|
"test_example_runs_bc_mean_field_perf"
|
|
"test_main_loop"
|
|
"test_clifford_circuit_2"
|
|
"test_decompose_specific_matrices"
|
|
"test_two_qubit_randomized_benchmarking"
|
|
"test_kak_decomposition_perf"
|
|
"test_example_runs_simon"
|
|
"test_decompose_random_unitary"
|
|
"test_decompose_size_special_unitary"
|
|
"test_api_retry_5xx_errors"
|
|
"test_xeb_fidelity"
|
|
"test_example_runs_phase_estimator_perf"
|
|
"test_cross_entropy_benchmarking"
|
|
];
|
|
|
|
meta = with lib; {
|
|
description = "A framework for creating, editing, and invoking Noisy Intermediate Scale Quantum (NISQ) circuits.";
|
|
homepage = "https://github.com/quantumlib/cirq";
|
|
changelog = "https://github.com/quantumlib/Cirq/releases/tag/v${version}";
|
|
license = licenses.asl20;
|
|
maintainers = with maintainers; [ drewrisinger ];
|
|
};
|
|
}
|