Skip to content
This repository was archived by the owner on Nov 7, 2024. It is now read-only.

Commit 55a7fcc

Browse files
authored
Z2charge (#516)
* add Z2Charge tests * lintinng * xRevert "lintinng" This reverts commit aad9a39. * added Z2Charge * added bbetter tests * linting * change argument order in U1Charge.random, add psuedo arguments to Z2Charge.random to keep signature consistent, fixed tests * fix typo * fix argument order * fix broken tests * import Z2Charge * import Z2Charge
1 parent 35d1247 commit 55a7fcc

9 files changed

Lines changed: 306 additions & 141 deletions

File tree

tensornetwork/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,4 +17,4 @@
1717
from tensornetwork import block_sparse
1818
from tensornetwork.block_sparse.blocksparsetensor import BlockSparseTensor, ChargeArray
1919
from tensornetwork.block_sparse.index import Index
20-
from tensornetwork.block_sparse.charge import U1Charge, BaseCharge
20+
from tensornetwork.block_sparse.charge import U1Charge, BaseCharge, Z2Charge

tensornetwork/backends/symmetric/decompositions_test.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,9 @@ def test_rq_decomposition(dtype, R, R1, num_charges):
164164
def test_qr_decomposition(dtype, R, R1):
165165
np.random.seed(10)
166166
D = 30
167-
charges = [U1Charge.random(-5, 5, D) for n in range(R)]
167+
charges = [
168+
U1Charge.random(dimension=D, minval=-5, maxval=5) for n in range(R)
169+
]
168170
flows = [True] * R
169171
A = BlockSparseTensor.random([Index(charges[n], flows[n]) for n in range(R)],
170172
dtype=dtype)

tensornetwork/block_sparse/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,4 @@
77
#pylint: disable=line-too-long
88
from tensornetwork.block_sparse.linalg import svd, qr, diag, sqrt, trace, inv, pinv, eye, zeros, ones, randn, eigh, eig, conj, reshape, transpose, random, norm
99
from tensornetwork.block_sparse.index import Index
10-
from tensornetwork.block_sparse.charge import U1Charge, BaseCharge
10+
from tensornetwork.block_sparse.charge import U1Charge, BaseCharge, Z2Charge

0 commit comments

Comments
 (0)