class_docstr: A :class:`~flash.core.data.io.output.Output` which returns the "preds" from the model outputs.
class_docstr_tok: ["A",":","class",":","`","~flash.core.data.io.output.Output","`","which","returns","the","``","preds","''","from","the","model","outputs","."]
code: def transform(self, sample: Any) -> Union[int, List[int]]:
      return (
          sample.get(DataKeys.PREDS, sample)
          if isinstance(sample, dict)
          else sample
      )
  
imports: from typing import Any, List, Union
  from flash.core.data.io.input import DataKeys
  from flash.core.data.io.output import Output
  from flash.core.registry import FlashRegistry
label_desc: Use image node_id 1 for calling the PredsOutput obj's underlying member method code with example usage: obj.transform(sample) and returns: unknown