fixed erreors detected by pylint

freude · Jul 26, 2020 · 20a84b8 · 20a84b8
1 parent deae048
commit 20a84b8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/nanonet/tb/hamiltonian.py b/nanonet/tb/hamiltonian.py
@@ -658,7 +658,11 @@ def get_hamiltonians_block_tridiagonal(self, left=None, right=None, optimized=Tr
             subblocks = split_into_subblocks(h0, left, right)
 
         h01, hl1, hr1 = cut_in_blocks(h0, subblocks)
-        hl1.append(hl[:left, -right:])
-        hr1.append(hr[-right:, :left])
+        if left is not None and right is not None:
+            hl1.append(hl[:left, -right:])
+            hr1.append(hr[-right:, :left])
+        else:
+            hl1.append(hl)
+            hr1.append(hr)
 
         return hl1, h01, hr1, subblocks
diff --git a/nanonet/tb/reduced_mode_space.py b/nanonet/tb/reduced_mode_space.py
@@ -135,8 +135,7 @@ def bs(E, h_l, h_0, h_r):
 
     """
 
-    h_list = [h_l, h_0 - E * np.identity(h_0.shape[0]), h_r]
-    vals, vects = surface_greens_function_poles(h_list)
+    vals, vects = surface_greens_function_poles(h_l, h_0 - E * np.identity(h_0.shape[0]), h_r)
     vals = np.diag(vals)
 
     vals_for_plot = vals.copy()