From 611be96195843d9f89c2a8a24d9dfc37ab4143ef Mon Sep 17 00:00:00 2001
From: Renato Mello <renato.msf@gmail.com>
Date: Mon, 18 Mar 2024 09:30:58 +0400
Subject: [PATCH] removing dill

---
 pyproject.toml     |   4 +-
 tests/test_dill.py | 127 ---------------------------------------------
 2 files changed, 2 insertions(+), 129 deletions(-)
 delete mode 100644 tests/test_dill.py

diff --git a/pyproject.toml b/pyproject.toml
index d0490b164e..2febb035ff 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -64,7 +64,6 @@ cirq = "^1.1.0"
 cvxpy = "^1.3.1"
 ply = "^3.11"
 scikit-learn = "^1.2.1"
-dill = "^0.3.6"
 pytest-cov = "^4.0.0"
 pylint = "^3.0.3"
 matplotlib = "^3.7.0"
@@ -102,7 +101,8 @@ qibotn = { git = "https://github.com/qiboteam/qibotn.git" }
 
 [tool.poetry.extras]
 tensorflow = ["tensorflow"]
-qinfo = ["cvxpy"]
+torch = ["torch"]
+qinfo = ["cvxpy", "stim"]
 
 [tool.pylint.reports]
 output-format = "colorized"
diff --git a/tests/test_dill.py b/tests/test_dill.py
deleted file mode 100644
index aaf312025f..0000000000
--- a/tests/test_dill.py
+++ /dev/null
@@ -1,127 +0,0 @@
-import pickle
-
-import dill
-import numpy as np
-import pytest
-
-
-def test_pickle_backend(backend):
-    serial = pickle.dumps(backend)
-    new_backend = pickle.loads(serial)
-    assert type(new_backend) == type(backend)
-    if hasattr(backend, "np"):
-        assert new_backend.np is backend.np
-
-
-def test_dill_backends(backend):
-    serial = dill.dumps(backend)
-    new_backend = dill.loads(serial)
-    assert type(new_backend) == type(backend)
-    assert new_backend.name == backend.name
-    assert new_backend.platform == backend.platform
-    assert type(new_backend.matrices) == type(backend.matrices)
-
-
-def test_dill_global_backend():
-    from qibo.backends import GlobalBackend
-
-    backend = GlobalBackend()
-    serial = dill.dumps(backend)
-    new_backend = dill.loads(serial)
-    assert type(new_backend) == type(backend)
-    assert new_backend.name == backend.name
-
-
-def test_dill_circuit(accelerators):
-    from qibo import Circuit, gates
-
-    circuit = Circuit(5, accelerators=accelerators)
-    circuit.add(gates.H(i) for i in range(5))
-    serial = dill.dumps(circuit)
-    new_circuit = dill.loads(serial)
-    assert type(new_circuit) == type(circuit)
-    assert new_circuit.nqubits == circuit.nqubits
-    assert new_circuit.to_qasm() == circuit.to_qasm()
-
-
-def test_dill_circuit_result(backend):
-    from qibo.models import QFT
-
-    circuit = QFT(4)
-    result = backend.execute_circuit(circuit)
-    serial = dill.dumps(result)
-    new_result = dill.loads(serial)
-    assert type(new_result) == type(result)
-    assert str(new_result) == str(result)
-    backend.assert_allclose(new_result, result)
-
-
-def test_dill_symbols():
-    from qibo.symbols import Symbol, X
-
-    matrix = np.random.random((2, 2))
-    s = Symbol(0, matrix)
-    x = X(1)
-    ns = dill.loads(dill.dumps(s))
-    nx = dill.loads(dill.dumps(x))
-    assert type(ns) == type(s)
-    assert type(nx) == type(x)
-    np.testing.assert_allclose(nx.matrix, x.matrix)
-    np.testing.assert_allclose(ns.matrix, s.matrix)
-
-
-def test_dill_measurement_symbol(backend):
-    from qibo import Circuit, gates
-
-    circuit = Circuit(1, density_matrix=True)
-    circuit.add(gates.H(0))
-    cresult = circuit.add(gates.M(0, collapse=True))
-    result = backend.execute_circuit(circuit, nshots=1)
-    symbol = cresult.symbols[0]
-    nsymbol = dill.loads(dill.dumps(symbol))
-    assert type(nsymbol) == type(symbol)
-    backend.assert_allclose(nsymbol.outcome(), symbol.outcome())
-
-
-def test_dill_hamiltonian(backend):
-    from qibo.hamiltonians import XXZ, Hamiltonian
-
-    matrix = np.random.random((4, 4))
-    ham1 = Hamiltonian(2, matrix, backend=backend)
-    ham2 = XXZ(3, backend=backend)
-    serial1 = dill.dumps(ham1)
-    serial2 = dill.dumps(ham2)
-    nham1 = dill.loads(serial1)
-    nham2 = dill.loads(serial2)
-    assert type(nham1) == type(ham1)
-    assert type(nham2) == type(ham2)
-    backend.assert_allclose(nham1.matrix, nham1.matrix)
-    backend.assert_allclose(nham2.matrix, nham2.matrix)
-
-
-def test_dill_symbolic_hamiltonian(backend):
-    from qibo.hamiltonians import SymbolicHamiltonian
-    from qibo.symbols import X, Y, Z
-
-    form = X(0) * X(1) + Y(0) * Y(1) + Z(0) * Z(1)
-    ham = SymbolicHamiltonian(form, backend=backend)
-    serial = dill.dumps(ham)
-    nham = dill.loads(serial)
-    assert type(nham) == type(ham)
-    backend.assert_allclose(nham.matrix, ham.matrix)
-
-
-def test_dill_variational_models(backend):
-    from qibo import gates
-    from qibo.hamiltonians import TFIM
-    from qibo.models import QAOA, VQE, Circuit
-
-    ham = TFIM(4, backend=backend)
-    circuit = Circuit(4)
-    circuit.add(gates.RX(i, theta=0) for i in range(4))
-    vqe = VQE(circuit, ham)
-    qaoa = QAOA(ham)
-    nvqe = dill.loads(dill.dumps(vqe))
-    nqaoa = dill.loads(dill.dumps(qaoa))
-    assert type(nvqe) == type(vqe)
-    assert type(nqaoa) == type(qaoa)