chore(deps): update to pybsm 0.10.2 and nrtk 0.18.0

Kitware · Jan 13, 2025 · 2f12ac2 · 2f12ac2
1 parent 28c9c00
commit 2f12ac2
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 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.2",
     "scikit-learn>=1.6.0",
     "smqtk_image_io",
     "tabulate",
@@ -42,7 +42,7 @@ dependencies = [
     "transformers",
     "datasets[vision]",
     "umap-learn",
-    "nrtk[headless]>=0.12.0",
+    "nrtk[headless]==0.18.0",
     "trame-annotations>=0.4.0",
 ]
 

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)
diff --git a/tests/test_transforms.py b/tests/test_transforms.py
@@ -14,6 +14,6 @@ def test_gaussian_blur():
 
 def test_pybsm():
     transforms = generate_transforms()
-    pybsm = transforms["nrtk_pybsm_2"]()
+    pybsm = transforms["nrtk_pybsm"]()
     pybsm.set_parameters({"D": 0.25, "f": 4.0})
     pybsm.execute(get_image())