Propagate attributes through in_cgs.

SWIFTSIM · Dec 5, 2024 · f3832b5 · f3832b5
1 parent f437231
commit f3832b5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/swiftsimio/objects.py b/swiftsimio/objects.py
@@ -882,6 +882,7 @@ def __setstate__(self, state):
     __getitem__ = _propagate_cosmo_array_attributes(unyt_array.__getitem__)
     __copy__ = _propagate_cosmo_array_attributes(unyt_array.__copy__)
     __deepcopy__ = _propagate_cosmo_array_attributes(unyt_array.__deepcopy__)
+    in_cgs = _propagate_cosmo_array_attributes(unyt_array.in_cgs)
     astype = _propagate_cosmo_array_attributes(unyt_array.astype)
     in_units = _propagate_cosmo_array_attributes(unyt_array.in_units)
     byteswap = _propagate_cosmo_array_attributes(unyt_array.byteswap)