ultracoldYEG / spinor-gpe

A python package simulating the quasi-2D pseudospin-1/2 Gross-Pitaevskii equation with NVIDIA GPU acceleration.
MIT License
4 stars 5 forks source link

Issue with potential energy #1

Open alexdgenu opened 3 years ago

alexdgenu commented 3 years ago

Hi,

I keep having this issue, and as I try to solve it persists (in google colab).

'TensorPropagator' object has no attribute 'pot_eng'

in this line (325):

pot = sum([d * pot for d, pot in zip(dens, ttools.to_numpy(self.pot_eng))])

bdsmith147 commented 2 years ago

TensorPropagator has an attribute called pot_eng_spin. Try replacing that in your version:

pot = sum([d * pot for d, pot in zip(dens, ttools.to_numpy(self.pot_eng_spin))])