From 9f39ea9ba4ac636a3c3e06bacb22063e9b729213 Mon Sep 17 00:00:00 2001 From: Alexey Taymanov Date: Wed, 15 Nov 2023 22:11:48 -0500 Subject: [PATCH] black --- src/learn_to_pick/pick_best.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/learn_to_pick/pick_best.py b/src/learn_to_pick/pick_best.py index 5b787d1..8ed4b36 100644 --- a/src/learn_to_pick/pick_best.py +++ b/src/learn_to_pick/pick_best.py @@ -134,7 +134,9 @@ def _generic_namespaces(context, actions): for a in actions: a["#"] = PickBestFeaturizer._generic_namespace(a) - def get_context_actions(self, event) -> Tuple[base.Featurized, List[base.Featurized]]: + def get_context_actions( + self, event + ) -> Tuple[base.Featurized, List[base.Featurized]]: context = base.embed(event.based_on or {}, self.model) to_select_from_var_name, to_select_from = next( iter(event.to_select_from.items()), (None, None)