This repository has been archived by the owner on Feb 1, 2024. It is now read-only.
generated from Lightning-Universe/Research-template
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
46 lines (36 loc) · 1.41 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
44
45
46
#!/usr/bin/env python
import os
from importlib.util import module_from_spec, spec_from_file_location
from pkg_resources import parse_requirements
from setuptools import find_packages, setup
_PATH_ROOT = os.path.dirname(__file__)
def _load_requirements(path_dir: str = _PATH_ROOT, file_name: str = "requirements.txt") -> list:
reqs = parse_requirements(open(os.path.join(path_dir, file_name)).readlines())
return list(map(str, reqs))
def _load_py_module(fname, pkg="research_app"):
spec = spec_from_file_location(os.path.join(pkg, fname), os.path.join(_PATH_ROOT, pkg, fname))
py = module_from_spec(spec)
spec.loader.exec_module(py)
return py
about = _load_py_module("__about__.py")
with open(os.path.join(_PATH_ROOT, "README.md"), encoding="utf-8") as fo:
readme = fo.read()
setup(
name="research-app",
version=about.__version__,
description=about.__docs__,
author=about.__author__,
author_email=about.__author_email__,
url=about.__homepage__,
download_url="https://github.com/PyTorchLightning/lightning",
license=about.__license__,
packages=find_packages(exclude=["tests", "docs"]),
long_description=readme,
long_description_content_type="text/markdown",
include_package_data=True,
zip_safe=False,
keywords=["deep learning", "pytorch", "AI"],
python_requires=">=3.8",
setup_requires=["wheel"],
install_requires=_load_requirements(),
)