fix __repr__

gau-nernst · gau-nernst · commit 4e585e9cc78c · 2024-06-25T20:15:56.000+08:00
diff --git a/torchao/prototype/quant_llm/quant_llm.py b/torchao/prototype/quant_llm/quant_llm.py
@@ -386,7 +386,7 @@ def dequantize(self, output_dtype=None):
     def __repr__(self):
         dtype = f"fp{1 + self.ebits + self.mbits}_e{self.ebits}m{self.mbits}"
         return (
-            f"{self.__class__.name}(dtype={dtype}, shape={self.shape}, "
+            f"{self.__class__.__name__}(dtype={dtype}, shape={self.shape}, "
             f"device={self.device}, requires_grad={self.requires_grad})"
         )
 

Original file line number	Diff line number	Diff line change
`@@ -386,7 +386,7 @@ def dequantize(self, output_dtype=None):`
`386`	`386`	`def __repr__(self):`
`387`	`387`	`dtype = f"fp{1 + self.ebits + self.mbits}_e{self.ebits}m{self.mbits}"`
`388`	`388`	`return (`
`389`		`- f"{self.__class__.name}(dtype={dtype}, shape={self.shape}, "`
	`389`	`+ f"{self.__class__.__name__}(dtype={dtype}, shape={self.shape}, "`
`390`	`390`	`f"device={self.device}, requires_grad={self.requires_grad})"`
`391`	`391`	`)`
`392`	`392`