Skip to content
3 changes: 1 addition & 2 deletions examples/scripts/3_Dynamics/3.10_Hybrid_swap_mc.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,9 +67,8 @@ class HybridSwapMCState(ts.SwapMCState, MDState):
last_swap: Last swap attempted
"""

last_permutation: torch.Tensor
_atom_attributes = (
ts.SwapMCState._atom_attributes | MDState._atom_attributes | {"last_permutation"} # noqa: SLF001
ts.SwapMCState._atom_attributes | MDState._atom_attributes # noqa: SLF001
)
_system_attributes = (
ts.SwapMCState._system_attributes | MDState._system_attributes # noqa: SLF001
Expand Down
6 changes: 4 additions & 2 deletions examples/tutorials/hybrid_swap_tutorial.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,9 +100,11 @@ class HybridSwapMCState(SwapMCState, MDState):
from MDState.
"""

last_permutation: torch.Tensor
_atom_attributes = (
MDState._atom_attributes | {"last_permutation"} # noqa: SLF001
ts.SwapMCState._atom_attributes | MDState._atom_attributes # noqa: SLF001
)
_system_attributes = (
ts.SwapMCState._system_attributes | MDState._system_attributes # noqa: SLF001
)


Expand Down
Loading