Update setup.py

aio-libs · Feb 1, 2019 · c6e53c9 · c6e53c9
1 parent cf1861d
commit c6e53c9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -11,7 +11,7 @@
 def get_version():
     regex = r"__version__\s=\s\'(?P<version>[\d\.ab]+?)\'"
 
-    path = 'aiocassandra.py'
+    path = 'aiocassandra/session.py'
 
     return re.search(regex, read(path)).group('version')
 
@@ -35,7 +35,7 @@ def read(*parts):
     setup_requires=['pytest-runner'] if needs_pytest else [],
     tests_require=['pytest', 'pytest-asyncio', 'pytest-cov'],
     python_requires='>=3.4.0',
-    py_modules=['aiocassandra'],
+    packages=['aiocassandra'],
     zip_safe=False,
     classifiers=[
         'Development Status :: 5 - Production/Stable',