Skip to content

Commit 0cff210

Browse files
committed
Removing class method
1 parent 6c3f4a6 commit 0cff210

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

torchao/utils.py

+1-4
Original file line numberDiff line numberDiff line change
@@ -573,7 +573,7 @@ class PlainAQTTensorImpl(...):
573573
__torch_function__ = classmethod(_dispatch__torch_function__)
574574
register_layout = classmethod(_register_layout)
575575
get_tensor_impl_constructor = classmethod(_get_tensor_impl_constructor)
576-
_get_to_kwargs = classmethod(_get_to_kwargs)
576+
_get_to_kwargs = _get_to_kwargs
577577

578578
def __tensor_flatten__(self):
579579
raise NotImplementedError("Subclasses must implement __tensor_flatten__")
@@ -587,9 +587,6 @@ def __tensor_unflatten__(
587587
def __repr__(self):
588588
raise NotImplementedError("Subclasses must implement __repr__")
589589

590-
def _apply_fn_to_data(self, fn):
591-
raise NotImplementedError("Subclasses must implement _apply_fn_to_data")
592-
593590
def get_layout(self):
594591
if not hasattr(self, "_layout"):
595592
return None

0 commit comments

Comments
 (0)