Update requirements

Make requirements less strict to remove (some) colab warnings

docs/conf.py

@@ -85,9 +85,9 @@
 # The theme to use for HTML and HTML Help pages.  See the documentation for
 # a list of builtin themes.
 #
-html_theme = 'alabaster'
+html_theme = "alabaster"
 # html_theme = 'classic'
-#html_theme = "sphinx_rtd_theme"
+# html_theme = "sphinx_rtd_theme"
 
 # Theme options are theme-specific and customize the look and feel of a theme
 # further.  For a list of options available for each theme, see the

requirements.txt

@@ -3,30 +3,27 @@ cython
 pre-commit
 PyYAML==5.1.1
 profane
-SQLAlchemy==1.3.5
+SQLAlchemy>=1.3.5
 sqlalchemy-utils
-psycopg2-binary==2.8.3
-matplotlib==3.1.1
+psycopg2-binary
+matplotlib
 pytorch-transformers==1.1.0
 colorlog==4.0.2
 # pytrec_eval with custom cutoffs is not yet on pypi
 git+git://github.com/cvangysel/pytrec_eval.git@a9fa75459934f6bbb7b142f272419305390e51ae
-pycapnp==0.6.4
 pytest==3.8.0
 pyjnius==1.2.1
-pescador==2.0.2
 pymagnitude==0.1.120
 h5py==2.9.0
 pytorch-pretrained-bert==0.4
 nltk==3.4.5
-pymongo==3.9.0
 # deps that the pymagnitude package isn't pulling in:
 lz4==2.1.10
 xxhash==1.3.0
 annoy==1.15.2
 fasteners==0.15
 #
-django==2.2.13
+#django==2.2.13
 pytest-mock==1.10.4
 mock
 pyserini==0.9.3.0

setup.py

@@ -34,33 +34,30 @@ def run(self):
     url="https://capreolus.ai",
     packages=setuptools.find_packages(),
     install_requires=[
-        "torch==1.2.0",
-        "torchvision==0.4.0",
+        "torch==1.4",
+        "torchvision==0.5",
         "cython",
         "pre-commit",
         "PyYAML==5.1.1",
         "profane",
-        "SQLAlchemy==1.3.5",
+        "SQLAlchemy",
         "sqlalchemy-utils",
-        "psycopg2-binary==2.8.3",
-        "matplotlib==3.1.1",
+        "psycopg2-binary",
+        "matplotlib",
         "pytorch-transformers==1.1.0",
         "colorlog==4.0.2",
         "pytrec_eval@git+git://github.com/cvangysel/pytrec_eval.git@a9fa75459934f6bbb7b142f272419305390e51ae#egg=pytrec_eval-0.4.1",
-        "pycapnp==0.6.4",
         "pytest",
         "pyjnius==1.2.1",
-        "pescador==2.0.2",
         "pymagnitude==0.1.120",
-        "h5py==2.9.0",
+        "h5py",
         "pytorch-pretrained-bert==0.4",
         "nltk==3.4.5",
-        "pymongo==3.9.0",
         "lz4==2.1.10",
         "xxhash==1.3.0",
         "annoy==1.15.2",
         "fasteners==0.15",
-        "django==2.2.13",
+        # "django==2.2.13",
         "pytest-mock==1.10.4",
         "mock",
         "pyserini==0.9.3.0",