File tree Expand file tree Collapse file tree 1 file changed +10
-10
lines changed
autoarray/dataset/imaging Expand file tree Collapse file tree 1 file changed +10
-10
lines changed Original file line number Diff line number Diff line change @@ -383,6 +383,16 @@ def apply_noise_scaling(
383
383
If True, the data values in the masked region are set to zero.
384
384
"""
385
385
386
+ if signal_to_noise_value is None :
387
+ noise_map = self .noise_map .native
388
+ noise_map [mask == False ] = noise_value
389
+ else :
390
+ noise_map = np .where (
391
+ mask == False ,
392
+ np .abs (self .data .native ) / signal_to_noise_value ,
393
+ self .noise_map .native ,
394
+ )
395
+
386
396
if should_zero_data :
387
397
data = np .where (np .invert (mask ), 0.0 , self .data .native )
388
398
else :
@@ -394,16 +404,6 @@ def apply_noise_scaling(
394
404
pixel_scales = self .data .pixel_scales ,
395
405
)
396
406
397
- if signal_to_noise_value is None :
398
- noise_map = self .noise_map .native
399
- noise_map [mask == False ] = noise_value
400
- else :
401
- noise_map = np .where (
402
- mask == False ,
403
- np .abs (data .native ) / signal_to_noise_value ,
404
- self .noise_map .native ,
405
- )
406
-
407
407
noise_map = Array2D .no_mask (
408
408
values = noise_map ,
409
409
shape_native = self .data .shape_native ,
You can’t perform that action at this time.
0 commit comments