opensportslib 0.1.1.dev4__tar.gz → 0.1.1.dev6__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.1.1.dev4
+Version: 0.1.1.dev6
 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
@@ -60,6 +60,10 @@ OpenSportsLib is designed for **researchers, ML engineers, and sports analytics
 
 ## Installation
 
+> Requires **Python 3.12+**.  
+> Supports CUDA 12.6 / 12.8 / 13.0 (with CPU fallback).  
+> PyTorch Geometric is supported up to PyTorch 2.10.*.
+
 ### Stable release
 
 ```bash
@@ -72,14 +76,22 @@ pip install opensportslib
 pip install --pre opensportslib
 ```
 
-### Optional extras
-
+### Setup Environment (PyTorch, CUDA aware & Optional Dependencies)
 ```bash
-pip install "opensportslib[localization]"
-pip install "opensportslib[py-geometric]" -f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html
-```
+# Install PyTorch (CPU/GPU auto-detected)
+opensportslib setup
+
+# Optional: install PyTorch Geometric support
+opensportslib setup --pyg
 
-> Requires **Python 3.12+**.
+# Optional: install for DALI support
+opensportslib setup --dali
+``` 
+---
+
+**Note:**  
+Run `opensportslib setup` to automatically configure dependencies.  
+If issues occur, manually install compatible versions of `torch`, `torchvision`, and related libraries according to your CUDA version or system compatibility.
 
 ---
 
@@ -204,13 +216,6 @@ cd opensportslib
 pip install -e .
 ```
 
-### With extras
-
-```bash
-pip install -e ".[localization]"
-pip install -e ".[py-geometric]" -f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html
-```
-
 ### Conda option
 
 If you prefer conda:
@@ -221,6 +226,18 @@ conda activate osl
 pip install -e .
 ```
 
+### Setup Environment (PyTorch, CUDA aware & Optional Dependencies)
+```bash
+# Install PyTorch (CPU/GPU auto-detected)
+opensportslib setup
+
+# Optional: install PyTorch Geometric support
+opensportslib setup --pyg
+
+# Optional: install for DALI support
+opensportslib setup --dali
+```
+
 ### Git workflow
 
 1. Make sure you are branching from `dev`

{opensportslib-0.1.1.dev4 → opensportslib-0.1.1.dev6}/README.md

@@ -29,6 +29,10 @@ OpenSportsLib is designed for **researchers, ML engineers, and sports analytics
 
 ## Installation
 
+> Requires **Python 3.12+**.  
+> Supports CUDA 12.6 / 12.8 / 13.0 (with CPU fallback).  
+> PyTorch Geometric is supported up to PyTorch 2.10.*.
+
 ### Stable release
 
 ```bash
@@ -41,14 +45,22 @@ pip install opensportslib
 pip install --pre opensportslib
 ```
 
-### Optional extras
-
+### Setup Environment (PyTorch, CUDA aware & Optional Dependencies)
 ```bash
-pip install "opensportslib[localization]"
-pip install "opensportslib[py-geometric]" -f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html
-```
+# Install PyTorch (CPU/GPU auto-detected)
+opensportslib setup
+
+# Optional: install PyTorch Geometric support
+opensportslib setup --pyg
 
-> Requires **Python 3.12+**.
+# Optional: install for DALI support
+opensportslib setup --dali
+``` 
+---
+
+**Note:**  
+Run `opensportslib setup` to automatically configure dependencies.  
+If issues occur, manually install compatible versions of `torch`, `torchvision`, and related libraries according to your CUDA version or system compatibility.
 
 ---
 
@@ -173,13 +185,6 @@ cd opensportslib
 pip install -e .
 ```
 
-### With extras
-
-```bash
-pip install -e ".[localization]"
-pip install -e ".[py-geometric]" -f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html
-```
-
 ### Conda option
 
 If you prefer conda:
@@ -190,6 +195,18 @@ conda activate osl
 pip install -e .
 ```
 
+### Setup Environment (PyTorch, CUDA aware & Optional Dependencies)
+```bash
+# Install PyTorch (CPU/GPU auto-detected)
+opensportslib setup
+
+# Optional: install PyTorch Geometric support
+opensportslib setup --pyg
+
+# Optional: install for DALI support
+opensportslib setup --dali
+```
+
 ### Git workflow
 
 1. Make sure you are branching from `dev`

{opensportslib-0.1.1.dev4 → opensportslib-0.1.1.dev6}/opensportslib/core/utils/data.py

@@ -57,10 +57,8 @@ def tracking_collate_fn(batch):
     except ImportError as exc:
         raise ImportError(
             "torch-geometric is required for tracking_collate_fn. "
-            "Install with: pip install \"opensportslib[py-geometric]\" "
-            "-f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html "
-            "or (editable): pip install -e \".[py-geometric]\" "
-            "-f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html"
+            "Run: `opensportslib setup --pyg` to install the correct version "
+            "based on your system (PyTorch & CUDA compatible)."
         ) from exc
     
     batch_size = len(batch)

{opensportslib-0.1.1.dev4 → opensportslib-0.1.1.dev6}/opensportslib/datasets/classification_dataset.py

@@ -500,11 +500,9 @@ class TrackingDataset(ClassificationDataset):
             from torch_geometric.data import Data
         except ImportError as exc:
             raise ImportError(
-                "torch-geometric is required for tracking_parquet datasets. "
-                "Install with: pip install \"opensportslib[py-geometric]\" "
-                "-f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html "
-                "or (editable): pip install -e \".[py-geometric]\" "
-                "-f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html"
+                "torch-geometric is required for tracking_collate_fn. "
+                "Run: `opensportslib setup --pyg` to install the correct version "
+                "based on your system (PyTorch & CUDA compatible)."
             ) from exc
 
         from opensportslib.datasets.utils.tracking import normalize_features
@@ -546,11 +544,9 @@ class TrackingDataset(ClassificationDataset):
             from torch_geometric.data import Data
         except ImportError as exc:
             raise ImportError(
-                "torch-geometric is required for tracking_parquet datasets. "
-                "Install with: pip install \"opensportslib[py-geometric]\" "
-                "-f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html "
-                "or (editable): pip install -e \".[py-geometric]\" "
-                "-f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html"
+                "torch-geometric is required for tracking_collate_fn. "
+                "Run: `opensportslib setup --pyg` to install the correct version "
+                "based on your system (PyTorch & CUDA compatible)."
             ) from exc
         
         from opensportslib.datasets.utils.tracking import (

{opensportslib-0.1.1.dev4 → opensportslib-0.1.1.dev6}/opensportslib/models/backbones/builder.py

@@ -39,11 +39,9 @@ from opensportslib.models.utils.shift import make_temporal_shift
 
 
 _PYG_IMPORT_ERROR_MSG = (
-    "torch-geometric is required for graph/tracking models. "
-    "Install with: pip install \"opensportslib[py-geometric]\" "
-    "-f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html "
-    "or (editable): pip install -e \".[py-geometric]\" "
-    "-f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html"
+    "torch-geometric is required for tracking_collate_fn. "
+    "Run: `opensportslib setup --pyg` to install the correct version "
+    "based on your system (PyTorch & CUDA compatible)."
 )
 
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.1.1.dev4
+Version: 0.1.1.dev6
 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
@@ -60,6 +60,10 @@ OpenSportsLib is designed for **researchers, ML engineers, and sports analytics
 
 ## Installation
 
+> Requires **Python 3.12+**.  
+> Supports CUDA 12.6 / 12.8 / 13.0 (with CPU fallback).  
+> PyTorch Geometric is supported up to PyTorch 2.10.*.
+
 ### Stable release
 
 ```bash
@@ -72,14 +76,22 @@ pip install opensportslib
 pip install --pre opensportslib
 ```
 
-### Optional extras
-
+### Setup Environment (PyTorch, CUDA aware & Optional Dependencies)
 ```bash
-pip install "opensportslib[localization]"
-pip install "opensportslib[py-geometric]" -f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html
-```
+# Install PyTorch (CPU/GPU auto-detected)
+opensportslib setup
+
+# Optional: install PyTorch Geometric support
+opensportslib setup --pyg
 
-> Requires **Python 3.12+**.
+# Optional: install for DALI support
+opensportslib setup --dali
+``` 
+---
+
+**Note:**  
+Run `opensportslib setup` to automatically configure dependencies.  
+If issues occur, manually install compatible versions of `torch`, `torchvision`, and related libraries according to your CUDA version or system compatibility.
 
 ---
 
@@ -204,13 +216,6 @@ cd opensportslib
 pip install -e .
 ```
 
-### With extras
-
-```bash
-pip install -e ".[localization]"
-pip install -e ".[py-geometric]" -f https://pytorch-geometric.com/whl/torch-2.10.0+cu128.html
-```
-
 ### Conda option
 
 If you prefer conda:
@@ -221,6 +226,18 @@ conda activate osl
 pip install -e .
 ```
 
+### Setup Environment (PyTorch, CUDA aware & Optional Dependencies)
+```bash
+# Install PyTorch (CPU/GPU auto-detected)
+opensportslib setup
+
+# Optional: install PyTorch Geometric support
+opensportslib setup --pyg
+
+# Optional: install for DALI support
+opensportslib setup --dali
+```
+
 ### Git workflow
 
 1. Make sure you are branching from `dev`

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opensportslib"
-version = "0.1.1.dev4"
+version = "0.1.1.dev6"
 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"