diff --git a/pyhealth/processors/sequence_processor.py b/pyhealth/processors/sequence_processor.py index 3f889051a..7877a8ebe 100644 --- a/pyhealth/processors/sequence_processor.py +++ b/pyhealth/processors/sequence_processor.py @@ -20,7 +20,7 @@ def __init__(self): self.code_vocab: Dict[Any, int] = {"": 0} self._next_index = 1 - @override + def fit(self, samples: List[Dict[str, Any]], field: str) -> None: for sample in samples: for token in sample[field]: @@ -32,7 +32,7 @@ def fit(self, samples: List[Dict[str, Any]], field: str) -> None: self.code_vocab[""] = len(self.code_vocab) - @override + def process(self, value: Any) -> torch.Tensor: """Process token value(s) into tensor of indices.