opensportslib 0.3.0.dev2__tar.gz → 0.3.0.dev3__tar.gz

{opensportslib-0.3.0.dev2/opensportslib.egg-info → opensportslib-0.3.0.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.3.0.dev2
+Version: 0.3.0.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

{opensportslib-0.3.0.dev2 → opensportslib-0.3.0.dev3}/opensportslib/core/config/loader.py

@@ -20,6 +20,10 @@ from .runtime_adapter import maybe_namespace, namespace_to_plain_dict
 _YAML_SUFFIXES = {".yaml", ".yml"}
 _TASK_DIRS = {"classification", "localization", "vqa"}
 _INTERPOLATION_RE = re.compile(r"\$\{([^}]+)\}")
+_CPU_OPENCV_SPLIT_TYPES = {
+    "VideoGameWithDali": "VideoGameWithOpencv",
+    "VideoGameWithDaliVideo": "VideoGameWithOpencvVideo",
+}
 
 
 def _load_single_yaml(path: str | Path) -> Any:
@@ -131,6 +135,42 @@ def load_raw_config(path: str | Path) -> dict[str, Any]:
     raise ValueError("Unsupported config format. Use YAML or JSON.")
 
 
+def _normalize_cpu_loader_backend(payload: dict[str, Any]) -> dict[str, Any]:
+    system = payload.get("SYSTEM", {})
+    if not isinstance(system, dict):
+        return payload
+
+    if str(system.get("device", "auto")).lower() != "cpu":
+        return payload
+
+    data = payload.get("DATA", {})
+    if not isinstance(data, dict):
+        return payload
+
+    common = data.get("common", {})
+    if not isinstance(common, dict):
+        return payload
+
+    runtime = common.get("runtime", {})
+    if not isinstance(runtime, dict):
+        runtime = {}
+        common["runtime"] = runtime
+    runtime["loader_backend"] = "opencv"
+
+    splits = common.get("splits", {})
+    if not isinstance(splits, dict):
+        return payload
+
+    for split_cfg in splits.values():
+        if not isinstance(split_cfg, dict):
+            continue
+        split_type = split_cfg.get("type")
+        if split_type in _CPU_OPENCV_SPLIT_TYPES:
+            split_cfg["type"] = _CPU_OPENCV_SPLIT_TYPES[split_type]
+
+    return payload
+
+
 def load_config(
     config_path: str | Path,
     *,
@@ -139,6 +179,7 @@ def load_config(
 ) -> Any:
     raw = load_raw_config(config_path)
     canonical = migrate_config(raw, as_namespace=False)
+    canonical = _normalize_cpu_loader_backend(canonical)
     assert_no_legacy_aliases(canonical)
 
     if validate:
@@ -168,6 +209,7 @@ def resolve_config(
 ) -> Any:
     payload = namespace_to_plain_dict(config)
     canonical = migrate_config(payload, as_namespace=False)
+    canonical = _normalize_cpu_loader_backend(canonical)
     assert_no_legacy_aliases(canonical)
     return maybe_namespace(canonical, as_namespace=as_namespace)
 

{opensportslib-0.3.0.dev2 → opensportslib-0.3.0.dev3/opensportslib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.3.0.dev2
+Version: 0.3.0.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

{opensportslib-0.3.0.dev2 → opensportslib-0.3.0.dev3}/pyproject.toml

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

{opensportslib-0.3.0.dev2 → opensportslib-0.3.0.dev3}/tests/test_config_architecture.py

@@ -2,7 +2,7 @@ from pathlib import Path
 
 import pytest
 
-from opensportslib.core.config import load_config, migrate_config, validate_config
+from opensportslib.core.config import load_config, migrate_config, resolve_config, validate_config
 from opensportslib.core.config.accessors import (
     get_vqa_xvars_feature_mode,
     get_xvars_feature_token_len_for_mode,
@@ -85,6 +85,20 @@ def test_localization_experiment_composes_all_layers():
     assert cfg["MODEL"]["components"]["video_encoder"]["source"]["name"] == "rny008_gsm"
 
 
+def test_resolve_config_cpu_normalizes_canonical_dali_localization_config():
+    cfg = load_config("opensportslib/configs/localization/video_dali.yaml", as_namespace=False)
+    cfg["SYSTEM"]["device"] = "cpu"
+
+    resolved = migrate_config(cfg, as_namespace=False)
+    resolved = validate_config(resolved)
+    resolved = resolve_config(resolved, as_namespace=False)
+
+    assert resolved["DATA"]["common"]["runtime"]["loader_backend"] == "opencv"
+    assert resolved["DATA"]["common"]["splits"]["train"]["type"] == "VideoGameWithOpencv"
+    assert resolved["DATA"]["common"]["splits"]["valid"]["type"] == "VideoGameWithOpencv"
+    assert resolved["DATA"]["common"]["splits"]["test"]["type"] == "VideoGameWithOpencvVideo"
+
+
 def test_vqa_xvars_experiment_composes_all_layers():
     cfg = load_config("opensportslib/configs/vqa/xvars.yaml", as_namespace=False)
 

{opensportslib-0.3.0.dev2 → opensportslib-0.3.0.dev3}/tests/test_config_split_override_sync.py

@@ -35,3 +35,11 @@ def test_resolve_config_keeps_canonical_split_sources():
     cfg = resolve_config(migrate_config(_legacy_localization_cfg(), as_namespace=False), as_namespace=False)
     assert cfg["DATA"]["common"]["splits"]["valid"]["annotation_path"].endswith("valid.json")
     assert cfg["DATA"]["common"]["splits"]["valid_data_frames"]["annotation_path"].endswith("valid.json")
+
+
+def test_resolve_config_cpu_forces_opencv_backend_and_split_types():
+    cfg = resolve_config(migrate_config(_legacy_localization_cfg(), as_namespace=False), as_namespace=False)
+    assert cfg["DATA"]["common"]["runtime"]["loader_backend"] == "opencv"
+    assert cfg["DATA"]["common"]["splits"]["train"]["type"] == "VideoGameWithOpencv"
+    assert cfg["DATA"]["common"]["splits"]["valid"]["type"] == "VideoGameWithOpencv"
+    assert cfg["DATA"]["common"]["splits"]["valid_data_frames"]["type"] == "VideoGameWithOpencvVideo"