minor code cleanup

Author: Thinh Nguyen

setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 from os import path
 
-pkg_name = [p for p in find_packages() if '.' not in p][0]
+pkg_name = next(p for p in find_packages() if '.' not in p)
 here = path.abspath(path.dirname(__file__))
 
 with open(path.join(here, 'README.md'), 'r') as f: