Added the warn_only member to RTStruct class

Brian-Taylor8 · Jan 8, 2023 · 6c6f24a · 6c6f24a
1 parent 02ec808
commit 6c6f24a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rt_utils/rtstruct.py b/rt_utils/rtstruct.py
@@ -12,9 +12,10 @@ class RTStruct:
     Wrapper class to facilitate appending and extracting ROI's within an RTStruct
     """
 
-    def __init__(self, series_data, ds: FileDataset, ROIGenerationAlgorithm=0):
+    def __init__(self, series_data, ds: FileDataset, ROIGenerationAlgorithm=0, warn_only: bool = False):
         self.series_data = series_data
         self.ds = ds
+        self.warn_only = warn_only
         self.frame_of_reference_uid = ds.ReferencedFrameOfReferenceSequence[
             -1
         ].FrameOfReferenceUID  # Use last strucitured set ROI