Skip to content

Commit

Permalink
naming cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
ataymano committed Nov 16, 2023
1 parent 9643dff commit 2f6d21e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/learn_to_pick/pick_best.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,7 +140,7 @@ def _generic_namespaces(context, actions):
for a in actions:
a["#"] = PickBestFeaturizer._generic_namespace(a)

def get_context_actions(
def get_context_and_actions(
self, event
) -> Tuple[base.Featurized, List[base.Featurized]]:
context = base.embed(event.based_on or {}, self.model)
Expand All @@ -162,7 +162,7 @@ def get_context_actions(
def featurize(
self, event: PickBestEvent
) -> Tuple[base.Featurized, List[base.Featurized], PickBestSelected]:
context, actions = self.get_context_actions(event)
context, actions = self.get_context_and_actions(event)

if self.auto_embed:
self._dotproducts(context, actions)
Expand Down

0 comments on commit 2f6d21e

Please sign in to comment.