Skip to content

Commit d0b6719

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 7343268 commit d0b6719

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

Diff for: machine_learning/mab.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -159,8 +159,7 @@ def select_arm(self) -> int:
159159
"""
160160
if self.total_counts < self.num_arms:
161161
return self.total_counts
162-
ucb_values = self.values + \
163-
np.sqrt(2 * np.log(self.total_counts) / self.counts)
162+
ucb_values = self.values + np.sqrt(2 * np.log(self.total_counts) / self.counts)
164163
return np.argmax(ucb_values)
165164

166165
def update(self, arm_index: int, reward: int) -> None:

0 commit comments

Comments
 (0)