diff --git a/test/NeurekaTestConf.py b/test/NeurekaTestConf.py index f878e68..a8f89a7 100644 --- a/test/NeurekaTestConf.py +++ b/test/NeurekaTestConf.py @@ -65,7 +65,7 @@ def check_valid_out_type(cls, v: IntegerType) -> IntegerType: @field_validator("weight_type") @classmethod def check_valid_weight_type(cls, v: IntegerType) -> IntegerType: - NeurekaTestConf._check_type("weight_type", v, ["int8"]) + NeurekaTestConf._check_type("weight_type", v, ["int8", "int7", "int6", "int5", "int4", "int3", "int2"]) return v @field_validator("scale_type")