diff --git a/micromagneticmodel/energy/dmi.py b/micromagneticmodel/energy/dmi.py index 17a89f4..cf5bd6a 100644 --- a/micromagneticmodel/energy/dmi.py +++ b/micromagneticmodel/energy/dmi.py @@ -8,6 +8,7 @@ @uu.inherit_docs @ts.typesystem( D=ts.Parameter(descriptor=ts.Scalar(), otherwise=df.Field), + suffix=ts.Typed(expected_type=str), crystalclass=ts.Subset( sample_set={ "Cnv", @@ -20,6 +21,16 @@ "D2d_x", "D2d_y", "D2d_z", + "S4_x", + "S4_y", + "S4_z", + "Dn_x", + "Dn_y", + "Dn_z", + "Cn_x", + "Cn_y", + "Cn_z", + "C2v", }, unpack=False, ), @@ -105,7 +116,7 @@ class DMI(EnergyTerm): """ - _allowed_attributes = ["D", "crystalclass"] + _allowed_attributes = ["D", "D1", "D2", "D3", "Dx", "Dy", "crystalclass", "suffix"] @property def _reprlatex(self):