From bf51fdc7eb306d18fef17552692bbf7fbe449a55 Mon Sep 17 00:00:00 2001 From: Francesco Conti Date: Sat, 27 Jul 2024 15:11:28 +0200 Subject: [PATCH] add support for < 8 bits in weight_type not tested end-to-end yet. --- test/NeurekaTestConf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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")