match original tsv orientation

Author: jpillai00

src/rbc/core/metrics/timeseries.py

@@ -167,10 +167,13 @@ def compute_timeseries(
     ts_path = out_dir / f"{stem}_timeseries.parquet"
     corr_path = out_dir / f"{stem}_connectome.parquet"
 
-    col_names = [str(label) for label in labels]
+    # ROIs
+    roi_names = [str(label) for label in labels]
+    # Timepoints
+    tp_names = [str(i) for i in range(ts.shape[1])]
 
-    pl.DataFrame(ts, schema=col_names).write_parquet(ts_path)
-    pl.DataFrame(corr, schema=col_names).write_parquet(corr_path)
+    pl.DataFrame(ts, schema=tp_names).write_parquet(ts_path)
+    pl.DataFrame(corr, schema=roi_names).write_parquet(corr_path)
 
     return TimeseriesOutputs(
         timeseries=ts_path,

tests/unit/core/test_timeseries.py

@@ -267,7 +267,7 @@ def test_round_trip(self, tmp_path: Path) -> None:
         assert len(result.labels) == 2
 
         ts_loaded = pl.read_parquet(result.timeseries)
-        assert ts_loaded.shape == (10, 2)
+        assert ts_loaded.shape == (2, 10)
 
         corr_loaded = pl.read_parquet(result.connectome)
         assert corr_loaded.shape == (2, 2)