setup.py

import sys
import setuptools

USERNAME = 'beasteers'
NAME = 'tflit'
VERSION = '0.1.2'


from tflit import tflite_install
if not (len(sys.argv) > 1 and sys.argv[1] == 'sdist'):
    # from importlib.machinery import SourceFileLoader
    # version = SourceFileLoader('tflit.tflite_install',
    #                            'tflit/tflite_install.py').load_module()
    tflite_install.check_install(verbose=True, upgrade=True)

setuptools.setup(
    name=NAME,
    version=VERSION,
    description='tflite_runtime, but easier.',
    long_description=open('README.md').read().strip(),
    long_description_content_type='text/markdown',
    author='Bea Steers',
    author_email='bea.steers@gmail.com',
    url='https://github.com/{}/{}'.format(USERNAME, NAME),
    packages=setuptools.find_packages(),
    package_data={NAME: ['*/*.tflite']},
    # entry_points={'console_scripts': ['{name}={name}:main'.format(name=NAME)]},
    install_requires=[
        'numpy',
        # 'tflite_runtime@{}'.format(tflite_install.get_tflite_url())
    ],
    extras_require={
        'tests': ['pytest-cov'],
    },
    license='MIT License',
    keywords='tflite runtime tensorflow keras deep machine learning model edge embedded compute cnn')