refactor: update pybsm version to 0.10.1

Kitware · Dec 16, 2024 · fc81cf0 · fc81cf0
1 parent 28c9c00
commit fc81cf0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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)