diff --git a/anylabeling/app_info.py b/anylabeling/app_info.py index 6d6a51f..1d529be 100644 --- a/anylabeling/app_info.py +++ b/anylabeling/app_info.py @@ -1,4 +1,4 @@ __appname__ = "AnyLabeling" __appdescription__ = "Effortless data labeling with AI support" -__version__ = "0.2.18" +__version__ = "0.2.19" __preferred_device__ = "CPU" # GPU or CPU diff --git a/setup.py b/setup.py index 8b0d284..7137c92 100644 --- a/setup.py +++ b/setup.py @@ -1,12 +1,9 @@ import os import re import platform -import logging from setuptools import find_packages, setup -package_name = "anylabeling" - def get_version(): """Get package version from app_info.py file""" @@ -34,9 +31,17 @@ def get_preferred_device(): return device +def get_package_name(): + """Get package name based on context""" + package_name = "anylabeling" + preferred_device = get_preferred_device() + if preferred_device == "GPU" and platform.system() != "Darwin": + package_name = "anylabeling-gpu" + return package_name + + def get_install_requires(): """Get python requirements based on context""" - global package_name install_requires = [ "imgviz>=0.11", "natsort>=7.1.0", @@ -56,11 +61,10 @@ def get_install_requires(): preferred_device = get_preferred_device() if preferred_device == "GPU" and platform.system() != "Darwin": install_requires.append("onnxruntime-gpu==1.14.1") - logging.info("Building AnyLabeling with GPU support") - package_name = "anylabeling-gpu" + print("Building AnyLabeling with GPU support") else: install_requires.append("onnxruntime==1.14.1") - logging.info("Building AnyLabeling without GPU support") + print("Building AnyLabeling without GPU support") if os.name == "nt": # Windows install_requires.append("colorama") @@ -76,7 +80,7 @@ def get_long_description(): setup( - name=package_name, + name=get_package_name(), version=get_version(), packages=find_packages(), description="Effortless data labeling with AI support",