setup.py

from setuptools import setup,find_packages
import sys, os

setup(name="lie-conv",
      description="Equivariant Convolutions on Lie Groups",
      version='0.1',
      author='Marc Finzi, Samuel Stanton, Pavel Izmailov',
      author_email='maf820@nyu.edu',
      license='MIT',
      python_requires='>=3.6',
      install_requires=['h5py',
      'olive-oil-ml @ git+https://github.com/mfinzi/olive-oil-ml',
      'torchdiffeq @ git+https://github.com/rtqichen/torchdiffeq'],#
      extras_require = {
          'GN':['torch-scatter','torch-sparse','torch-cluster','torch-geometric'],
          'TBX':['tensorboardX']
      },
      packages=find_packages(),#["oil",],#find_packages()
      long_description=open('README.md').read(),
)