cd ~/alicesw/O2DPG/UTILS/dfextensions/groupby_regression

# Stage files
git add groupby_regression_sliding_window.py
git add __init__.py
git add tests/test_groupby_regression_sliding_window.py

# Commit
git commit -m "feat: Phase 7 M7.1 - Sliding window regression

- Implement zero-copy accumulator for memory-efficient windowing
- Support 3D-6D sparse binned data with integer coordinates
- Integrate statsmodels (OLS, WLS, GLM, RLM fitters)
- Add comprehensive 31-test suite (30 passing, 1 skipped)
- Python 3.9.6 compatible with proper type hints

Key features:
- Hash-map based neighbor aggregation (no DataFrame replication)
- Boundary truncation for edge bins
- Quality flags for insufficient statistics
- Rich provenance metadata in .attrs
- Clear error messages for invalid inputs

Tests: 30/30 passing (1 skipped - v4 parity requires v4)
Performance: <5 min for 400k rows (numpy backend)
Version: 2.1.0

Ready for M7.1 review."

# Push
git push origin feature/groupby-optimization

Author: miranov25

UTILS/dfextensions/AliasDataFrame.py

@@ -3,7 +3,11 @@
 import numpy as np
 import json
 import uproot
-import ROOT  # type: ignore
+try:
+    import ROOT  # type: ignore
+except ImportError as e:
+    print(f"[AliasDataFrame] WARNING: ROOT import failed: {e}")
+    ROOT = None
 import matplotlib.pyplot as plt
 import networkx as nx
 import re
@@ -394,8 +398,8 @@ def _write_to_uproot(self, uproot_file, treename, dropAliasColumns):
         dtype_casts = {col: np.float32 for col in export_cols if self.df[col].dtype == np.float16}
         export_df = self.df[export_cols].astype(dtype_casts)
 
-        uproot_file[treename] = export_df
-
+        #uproot_file[treename] = export_df
+        uproot_file[treename] = {col: export_df[col].values for col in export_df.columns}
         for subframe_name, entry in self._subframes.items():
             entry["frame"].export_tree(uproot_file, f"{treename}__subframe__{subframe_name}", dropAliasColumns)
 

UTILS/dfextensions/AliasDataFrameTest.py

@@ -2,7 +2,7 @@
 import pandas as pd
 import numpy as np
 import os
-from AliasDataFrame import AliasDataFrame  # Adjust if needed
+from dfextensions.AliasDataFrame import AliasDataFrame  # Adjust if needed
 import tempfile
 
 class TestAliasDataFrame(unittest.TestCase):
@@ -218,6 +218,27 @@ def test_getattr_chained_subframe_access(self):
         expected = np.array([False, True, True])
         assert np.all(adf_main.sub.cutA == expected)  # explicit value check
 
+    def test_multi_column_index_join(self):
+        """Test subframe join with composite key (track_index, firstTFOrbit)"""
+        df_main = pd.DataFrame({
+            'track_index': [0, 0, 1, 1],
+            'firstTFOrbit': [100, 200, 100, 200],
+            'x': [1, 2, 3, 4]
+        })
+        df_sub = pd.DataFrame({
+            'track_index': [0, 0, 1, 1],
+            'firstTFOrbit': [100, 200, 100, 200],
+            'y': [10, 20, 30, 40]
+        })
+
+        adf_main = AliasDataFrame(df_main)
+        adf_sub = AliasDataFrame(df_sub)
+        adf_main.register_subframe("T", adf_sub, index_columns=["track_index", "firstTFOrbit"])
+        adf_main.add_alias("sum_xy", "x + T.y")
+        adf_main.materialize_alias("sum_xy")
+
+        expected = [11, 22, 33, 44]
+        np.testing.assert_array_equal(adf_main.df['sum_xy'].values, expected)
 
 if __name__ == "__main__":
     unittest.main()

UTILS/dfextensions/__init__.py

@@ -1,12 +1,19 @@
 # __init__.py
 
-from .AliasDataFrame import AliasDataFrame
+
 from .FormulaLinearModel import FormulaLinearModel
 from .DataFrameUtils import *  # if it provides general helper functions
-#from .groupby_regression import *  # or other relevant functions
+from .groupby_regression import *  # or other relevant functions
 
 __all__ = [
     "AliasDataFrame",
     "FormulaLinearModel",
     "GroupByRegressor"
 ]
+
+
+try:
+    from .AliasDataFrame import AliasDataFrame
+    __all__.append("AliasDataFrame")
+except ImportError:
+    pass