docstr: Calculate reward for each action based on context
  Args:
      context: task specific context
  
  Returns: action -> reward map
docstr_tok: ["Calculate","reward","for","each","action","based","on","context","Args",":","context",":","task","specific","context","Returns",":","action","-",">","reward","map"]
code: def _action_rewards(self, context) -> ActionRewards:
      """
      Calculate reward for each action based on context
      Args:
          context: task specific context
  
      Returns: action -> reward map
  
      """
      pass
  
imports: import logging
  import time
  from abc import ABC, abstractmethod
  from dataclasses import dataclass
  from typing import Optional, Sequence, Tuple, Union
  import numpy
  import torch
  from reagent.ope.estimators.estimator import Estimator, EstimatorResult
  from reagent.ope.estimators.types import Action, ActionDistribution, ActionSpace, Reward, Trainer, TrainingData, Values
  from reagent.ope.utils import Clamper, RunningAverage
  from torch import Tensor
label_desc: Use image node_id 1 for calling the BanditsModel obj's underlying member method code with example usage: obj._action_rewards(context) without return types