Skip to content

Commit a892641

Browse files
authored
[BUG] Adds test case for using csr_matrix with pickle - invalidates #2210 (#2843)
* Extend test parameter set to validate if csx_matrix works with pickle. * Invalidates bug #2210
1 parent ee15fda commit a892641

File tree

1 file changed

+16
-7
lines changed
  • aeon/transformations/collection/dictionary_based

1 file changed

+16
-7
lines changed

aeon/transformations/collection/dictionary_based/_sfa_fast.py

Lines changed: 16 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -747,13 +747,22 @@ def _get_test_params(cls, parameter_set="default"):
747747
`MyClass(**params)` or `MyClass(**params[i])` creates a valid test instance.
748748
"""
749749
# small window size for testing
750-
params = {
751-
"word_length": 4,
752-
"window_size": 4,
753-
"return_sparse": False,
754-
"feature_selection": "chi2",
755-
"alphabet_size": 2,
756-
}
750+
params = [
751+
{
752+
"word_length": 4,
753+
"window_size": 4,
754+
"return_sparse": False,
755+
"feature_selection": "chi2",
756+
"alphabet_size": 2,
757+
},
758+
{
759+
"word_length": 4,
760+
"window_size": 4,
761+
"return_sparse": True,
762+
"feature_selection": "chi2",
763+
"alphabet_size": 2,
764+
},
765+
]
757766
return params
758767

759768
def __getstate__(self):

0 commit comments

Comments
 (0)