-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
42 lines (33 loc) · 1.43 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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (c) 2015-2030, Zhi Liu. All rights reserved.
from os import path as os_path
import torchlib
from setuptools import setup
from setuptools import find_packages
from distutils.core import setup as cysetup
from Cython.Build import cythonize
this_dir = os_path.abspath(os_path.dirname(__file__))
def read_file(filename):
with open(os_path.join(this_dir, filename), encoding='utf-8') as f:
long_description = f.read()
return long_description
def read_requirements(filename):
return [line.strip() for line in read_file(filename).splitlines()
if not line.startswith('#')]
long_description = read_file('README.md'),
long_description_content_type = "text/markdown",
setup(name='torchlib',
version=torchlib.__version__,
description="PyTorch Tools for Synthetic Aperture Radar.",
author='Zhi Liu',
author_email='[email protected]',
url='https://iridescent.ink/torchlib/',
download_url='https://github.com/aisari/torchlib/',
license='MIT',
packages=find_packages(),
install_requires=read_requirements('requirements.txt'),
include_package_data=True,
keywords=['PyTorch', 'Synthetic Aperture Radar', 'Machine Learning', 'Deep Learning'],
ext_modules=cythonize([this_dir + '/torchlib/misc/draw_shapes.py', this_dir + '/torchlib/misc/mapping_operation.py', this_dir + '/torchlib/misc/sampling.py'])
)