We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c986e00 commit 5f23b8dCopy full SHA for 5f23b8d
1 file changed
pyrit/prompt_target/common/prompt_target.py
@@ -225,11 +225,7 @@ def supported_input_modalities(self) -> "set[PromptDataType]":
225
Returns:
226
set[PromptDataType]: Set of all individual modalities that appear in any supported combination.
227
"""
228
- all_modalities = set()
229
- supported_modalities = self.SUPPORTED_INPUT_MODALITIES # Works with both class attr and property
230
- for modality_set in supported_modalities:
231
- all_modalities.update(modality_set)
232
- return all_modalities
+ return {frozenset(s) for s in self.SUPPORTED_INPUT_MODALITIES}
233
234
@property
235
def supported_output_modalities(self) -> "set[PromptDataType]":
0 commit comments