diff --git a/tsml_eval/experiments/_get_classifier.py b/tsml_eval/experiments/_get_classifier.py index 808ae9f3..b390fab7 100644 --- a/tsml_eval/experiments/_get_classifier.py +++ b/tsml_eval/experiments/_get_classifier.py @@ -14,7 +14,7 @@ ["miniarsenal", "mini-arsenal"], ["multiarsenal", "multi-arsenal"], ["hydraclassifier", "hydra"], - ["multirockethydraclassifier", "multirockethydra", "multirocket-hydra"], + ["multirockethydraclassifier", "multirockethydra", "multirocket-hydra", "mrhydra"], ] deep_learning_classifiers = [ ["timecnnclassifier", "timecnn", "cnnclassifier", "cnn"], @@ -274,6 +274,7 @@ def _set_classifier_convolution_based( c == "multirockethydraclassifier" or c == "multirockethydra" or c == "multirocket-hydra" + or c == "mrhydra" ): from aeon.classification.convolution_based import MultiRocketHydraClassifier