diff --git a/pyproject.toml b/pyproject.toml index dbc3c30..11a0906 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -30,7 +30,7 @@ dependencies = [ "accelerate", "numpy", "Pillow", - "pybsm>=0.6,<=0.9.0", + "pybsm>=0.10.1", "scikit-learn>=1.6.0", "smqtk_image_io", "tabulate", diff --git a/src/nrtk_explorer/library/nrtk_transforms.yaml b/src/nrtk_explorer/library/nrtk_transforms.yaml index c63a8bd..71dcee9 100644 --- a/src/nrtk_explorer/library/nrtk_transforms.yaml +++ b/src/nrtk_explorer/library/nrtk_transforms.yaml @@ -115,10 +115,10 @@ nrtk_pybsm_detector_otf: type: float label: Focal length (m) -nrtk_pybsm_2: +nrtk_pybsm: perturber: nrtk.impls.perturb_image.pybsm.perturber.PybsmPerturber perturber_kwargs: nrtk_explorer.library.nrtk_transforms.create_sample_sensor_and_scenario - exec_default_args: [{ img_gsd: 0.15 }] + exec_default_args: [None, { img_gsd: 0.15 }] description: D: _path: [sensor, D] diff --git a/src/nrtk_explorer/library/yaml_transforms.py b/src/nrtk_explorer/library/yaml_transforms.py index e1f3365..3cef3c8 100644 --- a/src/nrtk_explorer/library/yaml_transforms.py +++ b/src/nrtk_explorer/library/yaml_transforms.py @@ -152,6 +152,6 @@ def execute(self, input, *input_args): input_args = self.exec_args input_array = np.asarray(input) - output_array = self._perturber.perturb(input_array, *input_args) + output_array, unknown = self._perturber.perturb(input_array, *input_args) return ImageModule.fromarray(output_array)