allow for a single text prompt for VL cosine map

bdaiinstitute · Aug 16, 2023 · c898a58 · c898a58
1 parent 49116cf
commit c898a58
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/zsos/mapping/value_map.py b/zsos/mapping/value_map.py
@@ -165,6 +165,8 @@ def get_value(point: np.ndarray) -> float:
                 pixel_value_within_radius(self.value_map[..., c], point_px, radius_px)
                 for c in range(self.value_channels)
             ]
+            if len(all_values) == 1:
+                return all_values[0]
             value = reduce_fn(all_values)
             return value