opensportslib 0.1.1.dev1__tar.gz → 0.1.1.dev3__tar.gz

{opensportslib-0.1.1.dev1/opensportslib.egg-info → opensportslib-0.1.1.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.1.1.dev1
+Version: 0.1.1.dev3
 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.dev3/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.dev1 → opensportslib-0.1.1.dev3}/opensportslib/core/utils/data.py

@@ -52,7 +52,16 @@ def tracking_collate_fn(batch):
     Custom collate function for tracking data.
     Uses PyG Batch.from_data_list for efficient C++ batching.
     """
-    from torch_geometric.data import Batch
+    try:
+        from torch_geometric.data import 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"
+        ) from exc
     
     batch_size = len(batch)
     seq_len = batch[0]['seq_len']
@@ -82,4 +91,4 @@ def mixup_data(x, y, alpha=0.2):
     index = torch.randperm(x.size(0)).to(x.device)
     mixed_x = lam * x + (1 - lam) * x[index]
     return mixed_x, y, y[index], lam
-    
+    

{opensportslib-0.1.1.dev1 → opensportslib-0.1.1.dev3}/opensportslib/datasets/classification_dataset.py

@@ -496,7 +496,16 @@ class TrackingDataset(ClassificationDataset):
         a copy of the feature array is made before augmentation and 
         normalization so the cached data is never mutated.
         """
-        from torch_geometric.data import Data
+        try:
+            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"
+            ) from exc
 
         from opensportslib.datasets.utils.tracking import normalize_features
         
@@ -533,7 +542,16 @@ class TrackingDataset(ClassificationDataset):
 
     def _getitem_on_the_fly(self, idx):
         """load, parse, and process a single sample from disk."""
-        from torch_geometric.data import Data
+        try:
+            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"
+            ) from exc
         
         from opensportslib.datasets.utils.tracking import (
             compute_deltas,
@@ -600,4 +618,4 @@ class TrackingDataset(ClassificationDataset):
         if label is not None:
             out["label"] = label
         return out
-        
+        

{opensportslib-0.1.1.dev1 → opensportslib-0.1.1.dev3}/opensportslib/models/backbones/builder.py

@@ -38,6 +38,23 @@ import os
 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"
+)
+
+
+def _import_pyg_nn():
+    try:
+        import torch_geometric.nn as pyg_nn
+    except ImportError as exc:
+        raise ImportError(_PYG_IMPORT_ERROR_MSG) from exc
+    return pyg_nn
+
+
 def _use_pretrained_weights(default=True):
     """Control pretrained backbone loading from env for offline/local runs."""
     flag = os.environ.get("OSL_PRETRAINED_WEIGHTS", "1").strip().lower()
@@ -350,7 +367,8 @@ class GraphEncoder(nn.Module):
     def __init__(self, input_dim, hidden_dim, num_layers, conv_type='gin', dropout=0.1):
         super().__init__()
 
-        from torch_geometric.nn import DeepGCNLayer
+        pyg_nn = _import_pyg_nn()
+        DeepGCNLayer = pyg_nn.DeepGCNLayer
         self.conv_type = conv_type
         self.feat_dim = hidden_dim
 
@@ -371,10 +389,14 @@ class GraphEncoder(nn.Module):
             self.gcn_layers.append(layer)
         
     def _build_conv_layer(self, conv_type, hidden_dim, dropout):
-        from torch_geometric.nn import (
-            GATv2Conv, EdgeConv, SAGEConv, 
-            GINConv, GENConv, GraphConv, MultiAggregation
-        )
+        pyg_nn = _import_pyg_nn()
+        GATv2Conv = pyg_nn.GATv2Conv
+        EdgeConv = pyg_nn.EdgeConv
+        SAGEConv = pyg_nn.SAGEConv
+        GINConv = pyg_nn.GINConv
+        GENConv = pyg_nn.GENConv
+        GraphConv = pyg_nn.GraphConv
+        MultiAggregation = pyg_nn.MultiAggregation
 
         if conv_type == 'graphconv':
             return GraphConv(
@@ -434,7 +456,8 @@ class GraphEncoder(nn.Module):
             raise ValueError(f"Unknown conv type: {conv_type}")
             
     def forward(self, x, edge_index, batch):
-        from torch_geometric.nn import global_mean_pool
+        pyg_nn = _import_pyg_nn()
+        global_mean_pool = pyg_nn.global_mean_pool
         
         x = self.node_encoder(x)
 
@@ -448,7 +471,8 @@ class GraphEncoder(nn.Module):
 
     def _compute_edge_dynamic(self, x, batch, k=8):
         """Compute dynamic KNN edges for EdgeConv layers."""
-        from torch_geometric.nn import knn_graph
+        pyg_nn = _import_pyg_nn()
+        knn_graph = pyg_nn.knn_graph
         edge_index = knn_graph(x, k=k, batch=batch, loop=False)
         return edge_index
 

{opensportslib-0.1.1.dev1 → opensportslib-0.1.1.dev3}/opensportslib/models/base/tracking.py

@@ -1,6 +1,5 @@
 import torch
 import torch.nn as nn
-from torch_geometric.nn import global_mean_pool
 
 from opensportslib.models.backbones.builder import build_backbone
 from opensportslib.models.neck.builder import build_neck
@@ -70,4 +69,4 @@ class TrackingModel(nn.Module):
         # classification
         logits = self.head(x)  # (B, num_classes)
         
-        return logits
+        return logits

opensportslib-0.1.1.dev3/opensportslib/setup/setup.py

@@ -0,0 +1,151 @@
+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
+    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.dev1 → opensportslib-0.1.1.dev3/opensportslib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.1.1.dev1
+Version: 0.1.1.dev3
 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.dev1 → opensportslib-0.1.1.dev3}/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.dev3/opensportslib.egg-info/entry_points.txt

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

{opensportslib-0.1.1.dev1 → opensportslib-0.1.1.dev3}/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.dev1 → opensportslib-0.1.1.dev3}/pyproject.toml

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