fix init bug

illinois-ceesd · Aug 5, 2024 · 7fa2c66 · 7fa2c66
1 parent 77740b3
commit 7fa2c66
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/y3prediction/unstart.py b/y3prediction/unstart.py
@@ -111,12 +111,10 @@ def __call__(self, dcoll, x_vec, eos, *, time=0.0):
         mass = eos.get_density(self._pres_bulk, self._temp_bulk,
                                self._y_bulk) + zeros
         velocity = self._vel_bulk
+        mom = mass*velocity
         energy = mass*(eos.get_internal_energy(self._temp_bulk, self._y_bulk)
                       + 0.5*np.dot(velocity, velocity))
 
-        mom = mass*velocity
-        energy = (energy + np.dot(mom, mom)/(2.0*mass))
-
         return make_conserved(
             dim=self._dim,
             mass=mass,