Removing class method

pytorch · Feb 5, 2025 · 0cff210 · 0cff210
1 parent 6c3f4a6
commit 0cff210
Showing 1 changed file with 1 addition and 4 deletions.
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 = classmethod(_get_to_kwargs)
+    _get_to_kwargs = _get_to_kwargs
 
     def __tensor_flatten__(self):
         raise NotImplementedError("Subclasses must implement __tensor_flatten__")
@@ -587,9 +587,6 @@ def __tensor_unflatten__(
     def __repr__(self):
         raise NotImplementedError("Subclasses must implement __repr__")
 
-    def _apply_fn_to_data(self, fn):
-        raise NotImplementedError("Subclasses must implement _apply_fn_to_data")
-
     def get_layout(self):
         if not hasattr(self, "_layout"):
             return None