From 36023a5f1c00c18ff87e139dbe84dd3d713ad5e8 Mon Sep 17 00:00:00 2001 From: shubhamugare Date: Fri, 11 Apr 2025 19:58:47 -0500 Subject: [PATCH] Fix a minor bug --- syncode/mask_store/mask_store.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/syncode/mask_store/mask_store.py b/syncode/mask_store/mask_store.py index 15ec1ef2..65cfdbc4 100644 --- a/syncode/mask_store/mask_store.py +++ b/syncode/mask_store/mask_store.py @@ -269,10 +269,10 @@ def _remove_left_whitespace( if (self._fsms.initial(fsm_state.terminal) == fsm_state or self._fsms.is_final(fsm_state)) and self._ignore_whitespace: if isinstance(remainder, bytes): # For bytes, use lstrip() to remove all leading whitespace - remainder = remainder.lstrip() + remainder = remainder.lstrip(b' ') elif isinstance(remainder, str): # For strings, use lstrip() to remove all leading whitespace - remainder = remainder.lstrip() + remainder = remainder.lstrip(' ') return remainder def _lookup_next_tokens_for_fsm_state(self, fsm_state: JointFSMState, next_terminal) -> torch.Tensor: