class_docstr: Abstract base class for UCB-style CB models.
class_docstr_tok: ["Abstract","base","class","for","UCB-style","CB","models","."]
docstr: This forward method will be called by the inference wrapper.
  By default it's same as regular forward(), but users can override it
      if they need special behavior in the inference wrapper.
docstr_tok: ["This","forward","method","will","be","called","by","the","inference","wrapper",".","By","default","it","'s","same","as","regular","forward","(",")",",","but","users","can","override","it","if","they","need","special","behavior","in","the","inference","wrapper","."]
code: def forward_inference(
      self, inp: torch.Tensor, ucb_alpha: Optional[float] = None
  ) -> Dict[str, torch.Tensor]:
      """
      This forward method will be called by the inference wrapper.
      By default it's same as regular forward(), but users can override it
          if they need special behavior in the inference wrapper.
      """
      return self.forward(inp, ucb_alpha=ucb_alpha)
  
file_path: ReAgent/reagent/models/cb_base_model.py
imports: import logging
  from abc import ABC, abstractmethod
  from typing import Dict, Optional
  import torch
label_desc: Use image node_id 4 for calling the UCBBaseModel obj's underlying member method code with example usage: obj.forward_inference(inp, ucb_alpha) and returns: self