File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed
Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -119,7 +119,7 @@ async def process_gain(
119119
120120class FractionationParameters (BaseModel ):
121121 fractionation : int
122- dose_per_frame : float
122+ dose_per_frame : Optional [ float ] = None
123123 num_frames : int = 0
124124 eer_path : Optional [str ] = None
125125 fractionation_file_name : str = "eer_fractionation.txt"
@@ -160,6 +160,7 @@ async def write_eer_fractionation_file(
160160 )
161161 ).all ()
162162 if session_parameters :
163+ fractionation_params .dose_per_frame = session_parameters [0 ].dose_per_frame
163164 fractionation_params .fractionation = session_parameters [0 ].eer_fractionation
164165 session_parameters [0 ].eer_fractionation_file = str (file_path )
165166 db .add (session_parameters [0 ])
@@ -168,6 +169,10 @@ async def write_eer_fractionation_file(
168169 if file_path .is_file ():
169170 return {"eer_fractionation_file" : str (file_path )}
170171
172+ if not fractionation_params .dose_per_frame :
173+ logger .error ("EER fractionation dose not set" )
174+ return {"eer_fractionation_file" : None }
175+
171176 if fractionation_params .num_frames :
172177 num_eer_frames = fractionation_params .num_frames
173178 elif (
You can’t perform that action at this time.
0 commit comments