def test_output_ndim(params):
    """Check that the number of dimensions is always 2."""
    transformer = WEASELMUSE(**params)
    ndim_actual = transformer.fit_transform(X, y).ndim
    assert ndim_actual == 2
Check that the number of dimensions is always 2.
pyts/multivariate/transformation/tests/test_weasel_muse.py