forked from learnables/learn2learn
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
43 lines (39 loc) · 1.19 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#!/usr/bin/env python3
import re
from setuptools import (
setup,
find_packages,
)
# Parses version number: https://stackoverflow.com/a/7071358
VERSIONFILE = 'learn2learn/_version.py'
verstrline = open(VERSIONFILE, "rt").read()
VSRE = r"^__version__ = ['\"]([^'\"]*)['\"]"
mo = re.search(VSRE, verstrline, re.M)
if mo:
VERSION = mo.group(1)
else:
raise RuntimeError('Unable to find version string in %s.' % (VERSIONFILE,))
# Installs the package
setup(
name='learn2learn',
packages=find_packages(),
version=VERSION,
description='PyTorch Meta-Learning Framework for Researchers',
long_description=open('README.md').read(),
long_description_content_type='text/markdown',
author='Debajyoti Datta, Ian bunner, Seb Arnold, Praateek Mahajan',
author_email='[email protected], [email protected]',
url='https://github.com/learnables/learn2learn',
download_url='https://github.com/learnables/learn2learn/archive/' + str(VERSION) + '.zip',
license='MIT',
classifiers=[],
scripts=[],
install_requires=[
'numpy>=1.15.4',
'gym>=0.14.0',
'torch>=1.1.0',
'torchvision>=0.3.0',
'pandas',
'requests',
],
)