From a83cae36521bdf4310e20aa00e7bea063443afa6 Mon Sep 17 00:00:00 2001 From: Ethan Wu Date: Tue, 21 May 2024 19:57:23 -0700 Subject: [PATCH] chore: reformat files --- mabby/__init__.py | 1 + mabby/stats.py | 1 + mabby/strategies/ucb.py | 1 + mabby/utils.py | 1 + 4 files changed, 4 insertions(+) diff --git a/mabby/__init__.py b/mabby/__init__.py index 06415d8..752a72b 100644 --- a/mabby/__init__.py +++ b/mabby/__init__.py @@ -8,6 +8,7 @@ * configure environments with custom arm spaces and rewards distributions * collect and visualize simulation metrics like regret and optimality """ + from mabby.agent import Agent from mabby.arms import Arm, BernoulliArm, GaussianArm from mabby.bandit import Bandit diff --git a/mabby/stats.py b/mabby/stats.py index dd336cf..aff5443 100644 --- a/mabby/stats.py +++ b/mabby/stats.py @@ -1,4 +1,5 @@ """Provides metric tracking for multi-armed bandit simulations.""" + from __future__ import annotations from collections.abc import Iterable diff --git a/mabby/strategies/ucb.py b/mabby/strategies/ucb.py index 9aa1d17..4f6fb83 100644 --- a/mabby/strategies/ucb.py +++ b/mabby/strategies/ucb.py @@ -1,4 +1,5 @@ """Provides implementations of upper confidence bound (UCB) strategies.""" + from __future__ import annotations import numpy as np diff --git a/mabby/utils.py b/mabby/utils.py index 9c39b32..79f973a 100644 --- a/mabby/utils.py +++ b/mabby/utils.py @@ -1,4 +1,5 @@ """Provides commonly used utility functions.""" + from __future__ import annotations import numpy as np