clone row_vector_cell

Author: curtischong

torch_sim/optimizers.py

@@ -936,6 +936,7 @@ def fire_init(
             cell_masses=cell_masses,
             # Optimization attributes
             reference_cell=state.cell.clone(),
+            row_vector_cell=state.row_vector_cell.clone(),
             cell_factor=cell_factor,
             pressure=pressure,
             dt=dt_start,
@@ -1020,7 +1021,6 @@ class FrechetCellFIREState(SimState, DeformGradMixin):
     stress: torch.Tensor
 
     # Optimization-specific attributes
-    reference_cell: torch.Tensor
     cell_factor: torch.Tensor
     pressure: torch.Tensor
     hydrostatic_strain: bool