diff --git a/deepparse/converter/data_padder.py b/deepparse/converter/data_padder.py index 642277cb..e21c32cb 100644 --- a/deepparse/converter/data_padder.py +++ b/deepparse/converter/data_padder.py @@ -15,7 +15,9 @@ class DataPadder: def __init__(self, padding_value: int = -100) -> None: self.padding_value = padding_value - def pad_word_embeddings_batch(self, batch: List[Tuple[List, List]], teacher_forcing: bool = False) -> Union[ + def pad_word_embeddings_batch( + self, batch: List[Tuple[List, List]], teacher_forcing: bool = False + ) -> Union[ Tuple[Tuple[torch.Tensor, List], torch.Tensor], Tuple[Tuple[torch.Tensor, List, torch.Tensor], torch.Tensor], ]: diff --git a/setup.py b/setup.py index 1cd42fe1..0719e3df 100644 --- a/setup.py +++ b/setup.py @@ -77,7 +77,7 @@ def main(): "fasttext-wheel", "pymagnitude-light", "poutyne", - "pandas==2.0.3", + "pandas", "urllib3", "cloudpathlib[s3, gs, azure]", ], @@ -94,7 +94,7 @@ def main(): "uvicorn==0.22.0", "sentry-sdk[fastapi]>=1.28.1", "python-decouple==3.8", - "black==23.3.0", # code formatting requirements + "black", # code formatting requirements "pylint==2.16.2", "pylint-django[with_django]==2.5.3", "pre-commit==3.3.3",