minor typo

Author: curtischong

torch_sim/state.py

@@ -47,7 +47,7 @@ class SimState:
             used by ASE.
         pbc (bool): Boolean indicating whether to use periodic boundary conditions
         atomic_numbers (torch.Tensor): Atomic numbers with shape (n_atoms,)
-        graph_idx(torch.Tensor, optional): Maps each atom index to its graph index.
+        graph_idx (torch.Tensor, optional): Maps each atom index to its graph index.
             Has shape (n_atoms,), defaults to None, must be unique consecutive
             integers starting from 0