Merge pull request #51 from songlab-cal/v0.4

v0.4

requirements.txt

@@ -1,4 +1,4 @@
-torch>=1.0
+torch>=1.0,<1.5
 tqdm
 tensorboardX
 scipy

setup.py

@@ -1,5 +1,18 @@
 # -*- coding: utf-8 -*-
 from setuptools import setup, find_packages
+import os
+
+
+def get_version():
+    directory = os.path.abspath(os.path.dirname(__file__))
+    init_file = os.path.join(directory, 'tape', '__init__.py')
+    with open(init_file) as f:
+        for line in f:
+            if line.startswith('__version__'):
+                delim = '"' if '"' in line else "'"
+                return line.split(delim)[1]
+        else:
+            raise RuntimeError("Unable to find version string.")
 
 
 with open('README.md', 'r') as rf:
@@ -8,27 +21,21 @@
 with open('LICENSE', 'r') as lf:
     LICENSE = lf.read()
 
+with open('requirements.txt', 'r') as reqs:
+    requirements = reqs.read().split()
+
 setup(
     name='tape_proteins',
     packages=find_packages(),
-    version='0.3',
+    version=get_version(),
     description="Repostory of Protein Benchmarking and Modeling",
     author="Roshan Rao, Nick Bhattacharya, Neil Thomas",
     author_email='[email protected], [email protected], [email protected]',
     url='https://github.com/songlab-cal/tape',
     license=LICENSE,
     keywords=['Proteins', 'Deep Learning', 'Pytorch', 'TAPE'],
     include_package_data=True,
-    install_requires=[
-        'torch>=1.0',
-        'tqdm',
-        'tensorboardX',
-        'scipy',
-        'lmdb',
-        'boto3',
-        'requests',
-        'biopython',
-    ],
+    install_requires=requirements,
     entry_points={
         'console_scripts': [
             'tape-train = tape.main:run_train',

tape/__init__.py

@@ -9,6 +9,8 @@
 import importlib
 import pkgutil
 
+__version__ = '0.4'
+
 
 # Import all the models and configs
 for _, name, _ in pkgutil.iter_modules([str(Path(__file__).parent / 'models')]):