Skip to content

Commit

Permalink
fixed a bug in setup.py
Browse files Browse the repository at this point in the history
  • Loading branch information
aliutkus committed Apr 23, 2019
1 parent 4c99434 commit 17d5427
Showing 1 changed file with 1 addition and 13 deletions.
14 changes: 1 addition & 13 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,16 +2,6 @@
from setuptools import setup
import torch.utils.cpp_extension as cpp

# To use a consistent encoding
from codecs import open
from os import path

here = path.abspath(path.dirname(__file__))

# Get the long description from the README file
with open(path.join(here, 'README.md'), encoding='utf-8') as f:
long_description = f.read()

# In any case, include the CPU version
modules = [cpp.CppExtension(
'torchsearchsorted.cpu',
Expand All @@ -29,13 +19,11 @@
name='torchsearchsorted',
version='1.0',
description='A searchsorted implementation for pytorch',
long_description=long_description,
long_description_content_type='text/markdown',
keywords='searchsorted',

author='Antoine Liutkus',
author_email='[email protected]',
packages=['torchsearchsorted torch'],
packages=['torchsearchsorted'],
ext_modules=modules,
cmdclass={
'build_ext': cpp.BuildExtension
Expand Down

0 comments on commit 17d5427

Please sign in to comment.