Works in streamlit offline with numba

multivariate_gaussian_overlap.py

@@ -28,6 +28,9 @@ from config import LOW_LEVEL_TASKS
 from multiprocessing import Pool, cpu_count
 import importlib
 
+# TEMP: force auto backend to skip GPU for debugging parity with CPU-only deployments.
+_FORCE_DISABLE_GPU_AUTO = True
+
 # Environment setup
 os.environ['PYDEVD_DISABLE_FILE_VALIDATION'] = '1'
 
@@ -67,6 +70,8 @@ _SENSOR_STATS_CACHE = {}
 
 def _import_optional_module(module_name: str):
     """Import a helper module regardless of package context."""
+    if module_name == 'gpu_overlap' and _FORCE_DISABLE_GPU_AUTO:
+        return None
     candidates = []
     if __package__:
         candidates.append(f"{__package__}.{module_name}")
@@ -509,18 +514,20 @@ def _compute_overlap_backend(means1_batch, vars1_batch, means2_batch, vars2_batc
     
     elif method == 'auto':
         # Auto fallback - try GPU → Numba → NumPy
-        gpu_module = _import_optional_module('gpu_overlap')
-        if gpu_module and getattr(gpu_module, 'GPU_AVAILABLE', False):
-            compute_overlap_batch_gpu = getattr(gpu_module, 'compute_overlap_batch_gpu', None)
-            if callable(compute_overlap_batch_gpu):
-                return compute_overlap_batch_gpu(
-                    means1_batch,
-                    vars1_batch,
-                    means2_batch,
-                    vars2_batch,
-                    tol,
-                    biomechanical_difference,
-                )
+        gpu_module = None
+        if not _FORCE_DISABLE_GPU_AUTO:
+            gpu_module = _import_optional_module('gpu_overlap')
+            if gpu_module and getattr(gpu_module, 'GPU_AVAILABLE', False):
+                compute_overlap_batch_gpu = getattr(gpu_module, 'compute_overlap_batch_gpu', None)
+                if callable(compute_overlap_batch_gpu):
+                    return compute_overlap_batch_gpu(
+                        means1_batch,
+                        vars1_batch,
+                        means2_batch,
+                        vars2_batch,
+                        tol,
+                        biomechanical_difference,
+                    )
         
         numba_module = _import_optional_module('numba_overlap')
         if numba_module and getattr(numba_module, 'NUMBA_AVAILABLE', False):
@@ -919,11 +926,13 @@ def _process_task_pair_batch(args):
     except ValueError as e:
         # Handle missing sensor errors gracefully
         if "Missing sensor data" in str(e):
+            warnings.warn(str(e), RuntimeWarning)
             return (t1, t2, None)
-        # For other ValueErrors, also return None but could log if needed
+        warnings.warn(f"ValueError while processing {t1} vs {t2}: {e}", RuntimeWarning)
         return (t1, t2, None)
-    except Exception:
+    except Exception as e:
         # Catch any other exceptions to prevent worker crash
+        warnings.warn(f"Unexpected error while processing {t1} vs {t2}: {e}", RuntimeWarning)
         return (t1, t2, None)