diff --git a/docs/changes/newsfragments/215.bugfix b/docs/changes/newsfragments/215.bugfix new file mode 100644 index 000000000..93d56dcf2 --- /dev/null +++ b/docs/changes/newsfragments/215.bugfix @@ -0,0 +1 @@ +Fix the output of :class:`junifer.markers.RSSETSMarker` to be 2D by `Synchon Mandal`_ \ No newline at end of file diff --git a/junifer/markers/ets_rss.py b/junifer/markers/ets_rss.py index c57f49eda..bb4eeef5d 100644 --- a/junifer/markers/ets_rss.py +++ b/junifer/markers/ets_rss.py @@ -133,6 +133,8 @@ class RSSETSMarker(BaseMarker): edge_ts, _ = _ets(out["data"]) # Compute the RSS out["data"] = np.sum(edge_ts**2, 1) ** 0.5 + # Make it 2D + out["data"] = out["data"][:, np.newaxis] # Set correct column label out["col_names"] = ["root_sum_of_squares_ets"] return out