Skip to content

Commit 038f960

Browse files
authored
Depreciate get_observer (#214)
* depreciate get_observer Signed-off-by: Kyle Sayers <kylesayrs@gmail.com> * use decorator, update docstring Signed-off-by: Kyle Sayers <kylesayrs@gmail.com> --------- Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent 6fe85e5 commit 038f960

File tree

2 files changed

+7
-8
lines changed

2 files changed

+7
-8
lines changed

src/compressed_tensors/quantization/quant_args.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818

1919
import torch
2020
from compressed_tensors.utils import Aliasable
21+
from compressed_tensors.utils.helpers import deprecated
2122
from pydantic import BaseModel, Field, field_validator, model_validator
2223

2324

@@ -123,12 +124,6 @@ class QuantizationArgs(BaseModel, use_enum_values=True):
123124
),
124125
)
125126

126-
def get_observer(self):
127-
"""
128-
:return: torch quantization FakeQuantize built based on these QuantizationArgs
129-
"""
130-
return self.observer
131-
132127
@field_validator("type", mode="before")
133128
def validate_type(cls, value) -> QuantizationType:
134129
if isinstance(value, str):
@@ -250,6 +245,10 @@ def pytorch_dtype(self) -> torch.dtype:
250245
else:
251246
raise ValueError(f"Invalid quantization type {self.type}")
252247

248+
@deprecated("QuantizationArgs.observer")
249+
def get_observer(self) -> str:
250+
return self.observer
251+
253252

254253
def round_to_quantized_type(
255254
tensor: torch.Tensor, args: QuantizationArgs

src/compressed_tensors/utils/helpers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -170,8 +170,8 @@ def deprecated(future_name: Optional[str] = None, message: Optional[str] = None)
170170
"""
171171
Decorator to mark functions as deprecated
172172
173-
:param new_function: Function called in place of depreciated function
174-
:param message: Depreciation message, replaces default depreciation message
173+
:param new_function: Function called in place of deprecated function
174+
:param message: Deprecation message, replaces default deprecation message
175175
"""
176176

177177
def decorator(func: Callable[[Any], Any]):

0 commit comments

Comments
 (0)