diff --git a/eppo_client/configuration_requestor.py b/eppo_client/configuration_requestor.py index 78a69e1..7dbb966 100644 --- a/eppo_client/configuration_requestor.py +++ b/eppo_client/configuration_requestor.py @@ -53,8 +53,8 @@ def store_flags(self, flag_data) -> Dict[str, Flag]: def store_bandits(self, bandit_data) -> Dict[str, BanditData]: bandit_configs = { - config["banditKey"]: BanditData(**config) - for config in cast(dict, bandit_data.get("bandits", [])) + key: BanditData(**data) + for key, data in cast(dict, bandit_data.get("bandits", {})).items() } self.__bandit_config_store.set_configurations(bandit_configs) return bandit_configs diff --git a/eppo_client/version.py b/eppo_client/version.py index d539d50..911557b 100644 --- a/eppo_client/version.py +++ b/eppo_client/version.py @@ -1 +1 @@ -__version__ = "3.1.1" +__version__ = "3.1.2"