Skip to content

Commit 3756332

Browse files
Override dose with session parameters table value for eer fractionation (#673)
1 parent db1671d commit 3756332

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/murfey/server/api/file_io_instrument.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ async def process_gain(
119119

120120
class 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 (

0 commit comments

Comments
 (0)