from setuptools import setup

import importlib

packages = [
    "mmcv",
    "cv2",
    "tqdm",
    "torch",
    "torchvision",
    "tensorrt",
    "numpy"
]

def is_package_installed(package_name):
    try:
        importlib.import_module(package_name)
        return True
    except:
        return False

for package_name in packages:
    if (not is_package_installed(package_name)):
        print(f"{package_name} is not installed !!!")
        
setup(name='cc_demo',
      version='0.0.1',
      packages=['models', 'projects'],
    )