reagent/mab/mab_algorithm.py [125:144]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def add_batch_observations(
        self,
        n_obs_per_arm: Tensor,
        sum_reward_per_arm: Tensor,
        sum_reward_squared_per_arm: Tensor,
        arm_ids: Optional[List[str]] = None,
    ):
        (
            n_obs_per_arm,
            sum_reward_per_arm,
            sum_reward_squared_per_arm,
        ) = reindex_multiple_tensors(
            all_ids=self.arm_ids,
            batch_ids=arm_ids,
            value_tensors=(
                n_obs_per_arm,
                sum_reward_per_arm,
                sum_reward_squared_per_arm,
            ),
        )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



reagent/mab/thompson_sampling.py [75:94]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def add_batch_observations(
        self,
        n_obs_per_arm: Tensor,
        sum_reward_per_arm: Tensor,
        sum_reward_squared_per_arm: Tensor,
        arm_ids: Optional[List[str]] = None,
    ):
        (
            n_obs_per_arm,
            sum_reward_per_arm,
            sum_reward_squared_per_arm,
        ) = reindex_multiple_tensors(
            all_ids=self.arm_ids,
            batch_ids=arm_ids,
            value_tensors=(
                n_obs_per_arm,
                sum_reward_per_arm,
                sum_reward_squared_per_arm,
            ),
        )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -