From 8808c0767a76905280cb2c70803d7adc5121717b Mon Sep 17 00:00:00 2001 From: Benjamin Bossan Date: Wed, 14 Aug 2024 11:43:25 +0200 Subject: [PATCH] TST Add LNTuningConfig and LoKrConfig to tests These two configs were missing in test_config.py. Also, reordered the list of all config classes to be sorted, which makes it easier to spot missing configs. --- tests/test_config.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/tests/test_config.py b/tests/test_config.py index 5e39769d12..cceee7b50a 100644 --- a/tests/test_config.py +++ b/tests/test_config.py @@ -29,7 +29,9 @@ FourierFTConfig, HRAConfig, IA3Config, + LNTuningConfig, LoHaConfig, + LoKrConfig, LoraConfig, MultitaskPromptTuningConfig, OFTConfig, @@ -47,21 +49,23 @@ PEFT_MODELS_TO_TEST = [("lewtun/tiny-random-OPTForCausalLM-delta", "v1")] ALL_CONFIG_CLASSES = ( - AdaptionPromptConfig, AdaLoraConfig, + AdaptionPromptConfig, + BOFTConfig, + FourierFTConfig, + HRAConfig, IA3Config, + LNTuningConfig, LoHaConfig, + LoKrConfig, LoraConfig, MultitaskPromptTuningConfig, + OFTConfig, + PolyConfig, PrefixTuningConfig, PromptEncoderConfig, PromptTuningConfig, - OFTConfig, - PolyConfig, - BOFTConfig, VeraConfig, - FourierFTConfig, - HRAConfig, )