opensportslib 0.1.1.dev2__tar.gz → 0.1.1.dev4__tar.gz

{opensportslib-0.1.1.dev2/opensportslib.egg-info → opensportslib-0.1.1.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.1.1.dev2
+Version: 0.1.1.dev4
 Summary: OpenSportsLib is the professional library, designed for advanced video understanding in sports. It provides state-of-the-art tools for action recognition, spotting, retrieval, and captioning, making it ideal for researchers, analysts, and developers working with sports video data.
 Author: Jeet Vora
 Requires-Python: >=3.12
@@ -22,17 +22,8 @@ Requires-Dist: opencv-python
 Requires-Dist: omegaconf
 Requires-Dist: timm
 Requires-Dist: seaborn
-Provides-Extra: localization
-Requires-Dist: nvidia-dali-cuda120; extra == "localization"
-Requires-Dist: cupy-cuda12x; extra == "localization"
-Requires-Dist: tabulate; extra == "localization"
-Requires-Dist: pytorch-lightning; extra == "localization"
-Provides-Extra: py-geometric
-Requires-Dist: torch-geometric; extra == "py-geometric"
-Requires-Dist: torch-scatter; extra == "py-geometric"
-Requires-Dist: torch-sparse; extra == "py-geometric"
-Requires-Dist: torch-cluster; extra == "py-geometric"
-Requires-Dist: torch-spline-conv; extra == "py-geometric"
+Requires-Dist: tabulate
+Requires-Dist: pytorch-lightning
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-cov; extra == "test"

opensportslib-0.1.1.dev4/opensportslib/cli.py

@@ -0,0 +1,22 @@
+import argparse
+from opensportslib.setup.setup import setup
+
+
+def main():
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument(
+        "command",
+        choices=["setup"]
+    )
+
+    parser.add_argument("--pyg", action="store_true")
+    parser.add_argument("--dali", action="store_true")
+
+    args = parser.parse_args()
+
+    if args.command == "setup":
+        setup(
+            pyg=args.pyg,
+            dali=args.dali  # reuse flag for simplicity
+        )

opensportslib-0.1.1.dev4/opensportslib/setup/setup.py

@@ -0,0 +1,152 @@
+import subprocess
+import sys
+
+
+CUDA_SUPPORT = [
+    "cu126",
+    "cu128",
+    "cu130",
+    "cpu"
+]
+
+def get_cuda_version():
+    try:
+        output = subprocess.check_output(["nvidia-smi"]).decode()
+
+        for line in output.split("\n"):
+            if "CUDA Version" in line:
+                ver  = line.split("CUDA Version:")[1].strip().split()[0]
+                print(f"CUDA Version found : {ver}")
+                return ver
+    except Exception:
+        return None
+
+
+CUDA_VERSION = get_cuda_version()
+
+def get_cpu_tag():
+    if not CUDA_VERSION:
+        return "cpu"
+
+def install_torch():
+    python = sys.executable
+    subprocess.call([python, "-m", "pip", "uninstall", "-y", "torch", "torchvision"])
+    for cuda in CUDA_SUPPORT:
+
+        print(f"\n Trying installation: {cuda}\n")
+        try:
+            if get_cpu_tag() == "cpu":
+                subprocess.check_call([
+                    python, "-m", "pip", "install",
+                    "torch", "torchvision", "torchaudio"
+                ])
+            else:
+
+                subprocess.check_call([
+                    python, "-m", "pip", "install",
+                    "torch", "torchvision", "torchaudio",
+                    "--index-url",
+                    f"https://download.pytorch.org/whl/{cuda}"
+                ])
+            print(f"\nSuccess with {cuda}")
+            return cuda
+
+        except Exception as e:
+            print(f"Failed with {cuda}: {e}")
+
+    raise RuntimeError("All CUDA installs failed")
+
+def install_dali():
+
+    python = sys.executable
+
+    print("\nInstalling dali extras...\n")
+
+    # DALI (only if GPU)
+    if CUDA_VERSION:
+        subprocess.check_call([
+            python, "-m", "pip", "install",
+            "nvidia-dali-cuda120"
+        ])
+    
+        # CuPy (CUDA-aware but auto-resolves internally)
+        subprocess.check_call([
+            python, "-m", "pip", "install",
+            "cupy-cuda12x"
+        ])
+
+def install_pyg():
+    import torch
+    from packaging import version
+
+    python = sys.executable
+    torch_version = "2.10.0" if version.parse(torch.__version__.split("+")[0]) > version.parse("2.10.0") else torch.__version__.split("+")[0]
+    cuda_tag = next((f"cu{CUDA_VERSION.replace('.', '')}" for _ in [0] if CUDA_VERSION), CUDA_SUPPORT[0])
+    cuda_tag = cuda_tag if cuda_tag in CUDA_SUPPORT else CUDA_SUPPORT[0]
+    print("\nInstalling Py-Geometric ecosystem...\n")
+    if cuda_tag == "cpu":
+        url =  f"https://data.pyg.org/whl/torch-{torch_version}+cpu.html"
+    else:
+        url = f"https://data.pyg.org/whl/torch-{torch_version}+{cuda_tag}.html"
+
+    subprocess.check_call([
+        python, "-m", "pip", "install",
+        "torch-geometric", "-f", url
+    ])
+    subprocess.check_call([
+        python, "-m", "pip", "install",
+        "torch-scatter", "-f", url
+    ])
+    subprocess.check_call([
+        python, "-m", "pip", "install",
+        "torch-sparse", "-f", url
+    ])
+    subprocess.check_call([
+        python, "-m", "pip", "install",
+        "torch-cluster", "-f", url
+    ])
+    subprocess.check_call([
+        python, "-m", "pip", "install",
+        "torch-spline-conv", "-f", url
+    ])
+
+def install_extras(dali=False, pyg=False):
+    if dali:
+        install_dali()
+        print("NVIDIA DALI installed successfully.")
+    if pyg:
+        install_pyg()
+        print("PyTorch Geometric installed successfully.")
+
+
+def verify():
+    import torch
+
+    print("\n Verifying installation...\n")
+    print("Torch:", torch.__version__)
+
+    if torch.cuda.is_available():
+        print("CUDA available")
+        print("GPU:", torch.cuda.get_device_name(0))
+    else:
+        print("Running on CPU")
+
+def setup(dali=False, pyg=False):
+    install_torch()
+    install_extras(dali=dali, pyg=pyg)
+    verify()
+
+
+# ----------------------------
+# CLI entry
+# ----------------------------
+if __name__ == "__main__":
+    import argparse
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--dali", action="store_true")
+    parser.add_argument("--pyg", action="store_true")
+
+    args = parser.parse_args()
+
+    setup(dali=args.dali, pyg=args.pyg)

{opensportslib-0.1.1.dev2 → opensportslib-0.1.1.dev4/opensportslib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.1.1.dev2
+Version: 0.1.1.dev4
 Summary: OpenSportsLib is the professional library, designed for advanced video understanding in sports. It provides state-of-the-art tools for action recognition, spotting, retrieval, and captioning, making it ideal for researchers, analysts, and developers working with sports video data.
 Author: Jeet Vora
 Requires-Python: >=3.12
@@ -22,17 +22,8 @@ Requires-Dist: opencv-python
 Requires-Dist: omegaconf
 Requires-Dist: timm
 Requires-Dist: seaborn
-Provides-Extra: localization
-Requires-Dist: nvidia-dali-cuda120; extra == "localization"
-Requires-Dist: cupy-cuda12x; extra == "localization"
-Requires-Dist: tabulate; extra == "localization"
-Requires-Dist: pytorch-lightning; extra == "localization"
-Provides-Extra: py-geometric
-Requires-Dist: torch-geometric; extra == "py-geometric"
-Requires-Dist: torch-scatter; extra == "py-geometric"
-Requires-Dist: torch-sparse; extra == "py-geometric"
-Requires-Dist: torch-cluster; extra == "py-geometric"
-Requires-Dist: torch-spline-conv; extra == "py-geometric"
+Requires-Dist: tabulate
+Requires-Dist: pytorch-lightning
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-cov; extra == "test"

{opensportslib-0.1.1.dev2 → opensportslib-0.1.1.dev4}/opensportslib.egg-info/SOURCES.txt

@@ -6,9 +6,11 @@ pyproject.toml
 examples/quickstart/basic_classification.py
 examples/quickstart/basic_localization.py
 opensportslib/__init__.py
+opensportslib/cli.py
 opensportslib.egg-info/PKG-INFO
 opensportslib.egg-info/SOURCES.txt
 opensportslib.egg-info/dependency_links.txt
+opensportslib.egg-info/entry_points.txt
 opensportslib.egg-info/requires.txt
 opensportslib.egg-info/top_level.txt
 opensportslib/apis/__init__.py
@@ -77,6 +79,7 @@ opensportslib/models/utils/impl/calf.py
 opensportslib/models/utils/impl/gsm.py
 opensportslib/models/utils/impl/gtad.py
 opensportslib/models/utils/impl/tsm.py
+opensportslib/setup/setup.py
 tests/conftest.py
 tests/test_config_utils_smoke.py
 tests/test_package_smoke.py

opensportslib-0.1.1.dev4/opensportslib.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+opensportslib = opensportslib.cli:main

{opensportslib-0.1.1.dev2 → opensportslib-0.1.1.dev4}/opensportslib.egg-info/requires.txt

@@ -12,22 +12,11 @@ opencv-python
 omegaconf
 timm
 seaborn
-
-[:platform_system != "Darwin" and platform_machine != "arm64"]
-decord
-
-[localization]
-nvidia-dali-cuda120
-cupy-cuda12x
 tabulate
 pytorch-lightning
 
-[py-geometric]
-torch-geometric
-torch-scatter
-torch-sparse
-torch-cluster
-torch-spline-conv
+[:platform_system != "Darwin" and platform_machine != "arm64"]
+decord
 
 [test]
 pytest

{opensportslib-0.1.1.dev2 → opensportslib-0.1.1.dev4}/pyproject.toml

@@ -4,19 +4,20 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opensportslib"
-version = "0.1.1.dev2"
+version = "0.1.1.dev4"
 description = "OpenSportsLib is the professional library, designed for advanced video understanding in sports. It provides state-of-the-art tools for action recognition, spotting, retrieval, and captioning, making it ideal for researchers, analysts, and developers working with sports video data."
 readme = "README.md"
 requires-python = ">=3.12"
-dependencies = [ "SoccerNet", "av", "decord; platform_system != 'Darwin' and platform_machine != 'arm64'", "evaluate", "scikit-learn", "torch", "torchvision", "transformers==4.57.3", "tokenizers==0.22.1", "accelerate", "wandb", "opencv-python", "omegaconf", "timm", "seaborn",]
+dependencies = [ "SoccerNet", "av", "decord; platform_system != 'Darwin' and platform_machine != 'arm64'", "evaluate", "scikit-learn", "torch", "torchvision", "transformers==4.57.3", "tokenizers==0.22.1", "accelerate", "wandb", "opencv-python", "omegaconf", "timm", "seaborn", "tabulate", "pytorch-lightning",]
 [[project.authors]]
 name = "Jeet Vora"
 
 [project.optional-dependencies]
-localization = [ "nvidia-dali-cuda120", "cupy-cuda12x", "tabulate", "pytorch-lightning",]
-py-geometric = [ "torch-geometric", "torch-scatter", "torch-sparse", "torch-cluster", "torch-spline-conv",]
 test = [ "pytest", "pytest-cov",]
 
+[project.scripts]
+opensportslib = "opensportslib.cli:main"
+
 [tool.setuptools.package-data]
 opensportslib = [ "opensportslib/config/*.yaml",]