import random


class BernouliArm:

    def __init__(self, s):
        self.s = s

    def pull(self):
        reward = 0
        if random.random() < self.s:
            reward = 1
        return reward

7ddaa444b7d29a7e1a85d8e528b7cfe06638221d
<mask token>


class BernouliArm:
    <mask token>
    <mask token>

<mask token>


class BernouliArm:

    def __init__(self, s):
        self.s = s
    <mask token>

<mask token>


class BernouliArm:

    def __init__(self, s):
        self.s = s

    def pull(self):
        reward = 0
        if random.random() < self.s:
            reward = 1
        return reward

import random


class BernouliArm:

    def __init__(self, s):
        self.s = s

    def pull(self):
        reward = 0
        if random.random() < self.s:
            reward = 1
        return reward
