[bugfix][QuantizationScheme] fix string that is read in tuple (#1693)

extra comma after string close is read as tuple since line break is made with parenthesis
neuralmagic · Jul 28, 2023 · 3ec2c16 · 3ec2c16
1 parent 9b4221e
commit 3ec2c16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparseml/pytorch/sparsification/quantization/quantization_scheme.py b/src/sparseml/pytorch/sparsification/quantization/quantization_scheme.py
@@ -81,7 +81,7 @@ class QuantizationArgs(BaseModel):
     strategy: str = Field(
         default="tensor",
         description=(
-            "scope of the quantization to be applied. can be 'tensor' or 'channel'",
+            "scope of the quantization to be applied. can be 'tensor' or 'channel'"
         ),
     )
     kwargs: Dict[str, Any] = Field(