Testing class method

pytorch · Feb 5, 2025 · 6c3f4a6 · 6c3f4a6
1 parent 457bdf1
commit 6c3f4a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchao/utils.py b/torchao/utils.py
@@ -573,7 +573,7 @@ class PlainAQTTensorImpl(...):
     __torch_function__ = classmethod(_dispatch__torch_function__)
     register_layout = classmethod(_register_layout)
     get_tensor_impl_constructor = classmethod(_get_tensor_impl_constructor)
-    _get_to_kwargs = _get_to_kwargs
+    _get_to_kwargs = classmethod(_get_to_kwargs)
 
     def __tensor_flatten__(self):
         raise NotImplementedError("Subclasses must implement __tensor_flatten__")