fix requirements

GRAAL-Research · Apr 14, 2024 · a82a904 · a82a904
1 parent 6f4e656
commit a82a904
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git 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
@@ -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",