pypi package fixes

facebookresearch · Aug 22, 2023 · 2b50b56 · 2b50b56 · ieew11233 · Sep 12, 2023
1 parent f84e177
commit 2b50b56
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 8 deletions.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -0,0 +1 @@
+include ./*.*
diff --git a/nougat/utils/__init__.py b/nougat/utils/__init__.py
diff --git a/predict.py b/predict.py
@@ -47,13 +47,6 @@ def get_args():
         default=None,
         help="Path to checkpoint directory",
     )
-    parser.add_argument(
-        "--pages",
-        type=int,
-        nargs="+",
-        default=None,
-        help="if only one PDF is processed, which pages? if 2 numbers are given, all in between will be processed.",
-    )
     parser.add_argument("--out", "-o", type=Path, help="Output directory.")
     parser.add_argument("--recompute", action="store_true")
     parser.add_argument("pdf", nargs="+", type=Path, help="PDF(s) to process.")

diff --git a/setup.cfg b/setup.cfg
@@ -0,0 +1,2 @@
+[metadata]
+description_file = README.md
diff --git a/setup.py b/setup.py
@@ -40,6 +40,7 @@ def read_long_description():
             "result",
         ]
     ),
+    py_modules=["predict","app","train","test"],
     python_requires=">=3.7",
     install_requires=[
         "transformers~=4.25.1",
@@ -49,7 +50,7 @@ def read_long_description():
         "datasets[vision]",
         "pytorch-lightning>=1.8.5",
         "nltk",
-        "python-Levenshtein-wheels",
+        "python-Levenshtein",
         "sentencepiece",
         "sconf>=0.2.3",
         "albumentations",