cSAXS/cdtools
0
0
Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-09-09 21:12:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

Rename probe_size parameter to probe_shape in FancyPtycho and MultislicePtycho classes to match the gaussian_probe function

Browse Source
This commit is contained in:
gnzng
2025-03-04 14:20:09 -08:00
parent db2450c5ae
commit efadaf807e
2 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/cdtools/models/fancy_ptycho.py
+2 -3
View File
@@ -195,7 +195,7 @@ class FancyPtycho(CDIModel):
@classmethod
def from_dataset(cls,
dataset,
probe_size=None,
probe_shape=None,
randomize_ang=0,
n_modes=1,
n_obj_modes=1,
@@ -277,7 +277,7 @@ class FancyPtycho(CDIModel):
)
# Finally, initialize the probe and object using this information
if probe_size is None:
if probe_shape is None:
probe = tools.initializers.SHARP_style_probe(
dataset,
propagation_distance=propagation_distance,
@@ -288,7 +288,6 @@ class FancyPtycho(CDIModel):
dataset,
obj_basis,
probe_shape,
probe_size,
propagation_distance=propagation_distance,
)
src/cdtools/models/multislice_ptycho.py
+2 -3
View File
@@ -181,7 +181,7 @@ class MultislicePtycho(CDIModel):
dataset,
dz,
nz,
probe_size=None,
probe_shape=None,
randomize_ang=0,
n_modes=1,
n_obj_modes=1,
@@ -262,7 +262,7 @@ class MultislicePtycho(CDIModel):
)
# Finally, initialize the probe and object using this information
if probe_size is None:
if probe_shape is None:
probe = tools.initializers.SHARP_style_probe(
dataset,
propagation_distance=propagation_distance,
@@ -273,7 +273,6 @@ class MultislicePtycho(CDIModel):
dataset,
obj_basis,
probe_shape,
probe_size,
propagation_distance=propagation_distance,
)