From 613814fe3fd1e8b7fae6899bee5cd9e4e26440d2 Mon Sep 17 00:00:00 2001 From: JalenCato Date: Wed, 30 Oct 2024 21:41:30 +0000 Subject: [PATCH] hard negative config --- .../graphstorm_processing/config/hard_negative_configs.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/graphstorm-processing/graphstorm_processing/config/hard_negative_configs.py b/graphstorm-processing/graphstorm_processing/config/hard_negative_configs.py index c4924c57e0..00a99c8f1b 100644 --- a/graphstorm-processing/graphstorm_processing/config/hard_negative_configs.py +++ b/graphstorm-processing/graphstorm_processing/config/hard_negative_configs.py @@ -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() \ No newline at end of file