temp custom nodes

PennyLaneAI · Sep 20, 2024 · 0d49388 · 0d49388
1 parent b4a0f9c
commit 0d49388
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pennylane/spin/spin_hamiltonian.py b/pennylane/spin/spin_hamiltonian.py
@@ -389,7 +389,7 @@ def kitaev(n_cells, coupling=None, boundary_condition=False):
     return hamiltonian.simplify()
 
 
-def spin_hamiltonian(lattice):
+def spin_hamiltonian(lattice, custom_nodes=None):
     r"""Generates a spin Hamiltonian for a custom lattice.
 
     Args:
@@ -440,4 +440,10 @@ def spin_hamiltonian(lattice):
 
         hamiltonian += coeff * (opmap[op1](v1) @ opmap[op2](v2))
 
+    for node in custom_nodes:
+        n = node[0]
+        op = node[1]
+        coeff = node[2]
+        hamiltonian += coeff * opmap[op](n)
+
     return hamiltonian.simplify()