add contours_lines and classifiers to __attrs__ in RollPass (#243)

pyroll-project · Aug 23, 2024 · 833dd29 · 833dd29
1 parent 8f6afb6
commit 833dd29
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pyroll/core/roll_pass/roll_pass.py b/pyroll/core/roll_pass/roll_pass.py
@@ -236,6 +236,13 @@ class InProfile(Profile, DiskElementUnit.DiskElement.InProfile, DeformationUnit.
         class OutProfile(Profile, DiskElementUnit.DiskElement.OutProfile, DeformationUnit.OutProfile):
             """Represents an outgoing profile of a disk element unit."""
 
+    @property
+    def __attrs__(self):
+        return super().__attrs__ | {
+            "contour_lines": self.contour_lines,
+            "classifiers": self.classifiers,
+        }
+
     def _get_oriented_geom(self, geom):
         orientation = self.orientation