From 555e39b5cbc1199ec8410298c5b1e7e5198d94ec Mon Sep 17 00:00:00 2001 From: Simon Cross Date: Mon, 25 Oct 2021 17:50:48 +0200 Subject: [PATCH] Fix definition of site dims in plotting functions. --- qutip/lattice.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/qutip/lattice.py b/qutip/lattice.py index e39376918a..aaec62cf20 100644 --- a/qutip/lattice.py +++ b/qutip/lattice.py @@ -7,7 +7,7 @@ try: import matplotlib.pyplot as plt -except: +except ImportError: pass @@ -1052,8 +1052,7 @@ def display_unit_cell(self, label_on=False): Qin[i, j] = self._H_intra[ i0*self._length_for_site+i, j0*self._length_for_site+j] - dim_site = list(np.delete(self.cell_tensor_config, [0], None)) - dims_site = [dim_site, dim_site] + dims_site = [self.cell_site_dof, self.cell_site_dof] Hcell[i0][j0] = Qobj(Qin, dims=dims_site) fig = plt.figure(figsize=[CNS*2, CNS*2.5]) @@ -1152,8 +1151,7 @@ def display_lattice(self): Qin[i, j] = self._H_intra[ i0*self._length_for_site+i, j0*self._length_for_site+j] - dim_site = list(np.delete(self.cell_tensor_config, [0], None)) - dims_site = [dim_site, dim_site] + dims_site = [self.cell_site_dof, self.cell_site_dof] Hcell[i0][j0] = Qobj(Qin, dims=dims_site) j0 = 0 @@ -1242,6 +1240,5 @@ def display_lattice(self): plt.axis('off') plt.show() plt.close() - dim_site = list(np.delete(self.cell_tensor_config, [0], None)) - dims_site = [dim_site, dim_site] + dims_site = [self.cell_site_dof, self.cell_site_dof] return Qobj(inter_T, dims=dims_site)