add support for < 8 bits in weight_type

not tested end-to-end yet.
pulp-platform · Jul 27, 2024 · bf51fdc · bf51fdc
1 parent 67113b3
commit bf51fdc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")