Skip to content

Commit c8ac33f

Browse files
committed
Add Parameter.SegmentationMap annotation method
1 parent c634743 commit c8ac33f

File tree

2 files changed

+20
-1
lines changed

2 files changed

+20
-1
lines changed

Changelog.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
## 0.0.88
2-
*INCOMPLETE*
2+
+ Added `Parameter.SegmentationMap` annotation method to specify segmentation maps.
33

44
## 0.0.87
55
+ Added support for `mp3` respose format in `muna.beta.openai.audio.speech.create` method.

muna/types/parameter.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -198,4 +198,23 @@ def DepthMap(
198198
description=description,
199199
denotation="depth_map",
200200
**kwargs
201+
)
202+
203+
@classmethod
204+
def SegmentationMap(
205+
cls,
206+
*,
207+
description: str,
208+
**kwargs
209+
) -> Parameter:
210+
"""
211+
Segmentation map parameter.
212+
213+
This must be a boolean tensor with shape (H,W) or (N,H,W).
214+
"""
215+
return Parameter(
216+
name="",
217+
description=description,
218+
denotation="segmentation_map",
219+
**kwargs
201220
)

0 commit comments

Comments
 (0)