Skip to content

Commit

Permalink
hard negative config
Browse files Browse the repository at this point in the history
  • Loading branch information
jalencato committed Oct 30, 2024
1 parent ba88b9f commit 613814f
Showing 1 changed file with 2 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -30,13 +30,9 @@ class HardNegativeConfig(FeatureConfig):

def __init__(self, config: Mapping):
super().__init__(config)
self.separator = self._transformation_kwargs.get("separator")
self.separator = self._transformation_kwargs.get("separator", None)

self._sanity_check()

def _sanity_check(self) -> None:
super()._sanity_check()
assert self.action in [
HUGGINGFACE_TOKENIZE,
HUGGINGFACE_EMB,
], f"huggingface action needs to be one of {HUGGINGFACE_TOKENIZE, HUGGINGFACE_EMB}"
super()._sanity_check()

0 comments on commit 613814f

Please sign in to comment.