io4it 0.0.0.9__tar.gz → 0.0.0.11__tar.gz

{io4it-0.0.0.9 → io4it-0.0.0.11}/PKG-INFO

@@ -1,8 +1,9 @@
 Metadata-Version: 2.1
 Name: io4it
-Version: 0.0.0.9
+Version: 0.0.0.11
 Home-page: 
 Author: 
 Author-email: 
 Keywords: orange3 add-on
+Provides-Extra: cuda
 License-File: License.txt

{io4it-0.0.0.9 → io4it-0.0.0.11}/io4it.egg-info/PKG-INFO

@@ -1,8 +1,9 @@
 Metadata-Version: 2.1
 Name: io4it
-Version: 0.0.0.9
+Version: 0.0.0.11
 Home-page: 
 Author: 
 Author-email: 
 Keywords: orange3 add-on
+Provides-Extra: cuda
 License-File: License.txt

{io4it-0.0.0.9 → io4it-0.0.0.11}/io4it.egg-info/requires.txt

@@ -8,3 +8,8 @@ pyannote.audio
 pyannote.core
 wave
 scikit-learn
+
+[cuda]
+torch
+torchvision
+torchaudio

{io4it-0.0.0.9 → io4it-0.0.0.11}/setup.py

@@ -1,10 +1,12 @@
 from setuptools import setup, find_packages
+import platform
+import warnings
 
 # Nom du package PyPI ('pip install NAME')
 NAME = "io4it"
 
 # Version du package PyPI
-VERSION = "0.0.0.9"  # la version doit être supérieure à la précédente sinon la publication sera refusée
+VERSION = "0.0.0.11"  # la version doit être supérieure à la précédente sinon la publication sera refusée
 
 # Facultatif / Adaptable à souhait
 AUTHOR = ""
@@ -33,6 +35,16 @@ PACKAGE_DATA = {
 # Dépendances
 INSTALL_REQUIRES = ["boto3", "docling", "docling-core", "speechbrain", "whisper", "whisper-openai", "pyannote.audio", "pyannote.core", "wave", "scikit-learn"]
 
+
+
+# Extras optionnels pour CUDA
+EXTRAS_REQUIRE = {
+    "cuda": [
+        "torch",
+        "torchvision",
+        "torchaudio"
+    ]
+}
 # Spécifie le dossier contenant les widgets et le nom de section qu'aura l'addon sur Orange
 ENTRY_POINTS = {
     "orange.widgets": (
@@ -43,17 +55,31 @@ ENTRY_POINTS = {
 
 NAMESPACE_PACKAGES = ["orangecontrib"]
 
-setup(name=NAME,
-      version=VERSION,
-      author=AUTHOR,
-      author_email=AUTHOR_EMAIL,
-      url=URL,
-      description=DESCRIPTION,
-      license=LICENSE,
-      keywords=KEYWORDS,
-      packages=PACKAGES,
-      package_data=PACKAGE_DATA,
-      install_requires=INSTALL_REQUIRES,
-      entry_points=ENTRY_POINTS,
-      namespace_packages=NAMESPACE_PACKAGES,
-      )
+
+
+#   Message informatif si on est sous Windows
+if platform.system() == "Windows":
+    warnings.warn(
+        "Vous êtes sur Windows. Si vous avez une carte NVIDIA CUDA, installez avec :\n"
+        "   pip install io4it[cuda] --extra-index-url https://download.pytorch.org/whl/cu126",
+        UserWarning
+    )
+
+
+setup(
+    name=NAME,
+    version=VERSION,
+    author=AUTHOR,
+    author_email=AUTHOR_EMAIL,
+    url=URL,
+    description=DESCRIPTION,
+    license=LICENSE,
+    keywords=KEYWORDS,
+    packages=PACKAGES,
+    package_data=PACKAGE_DATA,
+    install_requires=INSTALL_REQUIRES,
+    extras_require=EXTRAS_REQUIRE,
+    entry_points=ENTRY_POINTS,
+    namespace_packages=NAMESPACE_PACKAGES,
+)
+