Uploaded image for project: 'Beam'
  1. Beam
  2. BEAM-13970 RunInference V1
  3. BEAM-14362

Add flag to drop example from PredicitonResult

Details

    • Sub-task
    • Status: Open
    • P2
    • Resolution: Unresolved
    • None
    • None
    • sdk-py-core

    Description

      RunInference currently returns both the example/input and prediction. 

      PredictionResult:
        example: torch(...)
        inference: torch(...)
      

      Users may want the ability to only return the inference to minimize potential memory/serialization issues later in the pipeline. They can do this with a flag, and the return value may look like

      PredictionResult:
        example: None
        inference: torch(...)
      

       

      Attachments

        Activity

          People

            Unassigned Unassigned
            yeandy Andy Ye
            Votes:
            0 Vote for this issue
            Watchers:
            2 Start watching this issue

            Dates

              Created:
              Updated: