Skip to content

Commit f3ede47

Browse files
committed
remove precision property from ttp and add deprecation message
1 parent 4d0e6bd commit f3ede47

File tree

2 files changed

+21
-7
lines changed

2 files changed

+21
-7
lines changed

pytorch_lightning/accelerators/accelerator.py

Lines changed: 21 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
from pytorch_lightning.plugins.precision import ApexMixedPrecisionPlugin, NativeMixedPrecisionPlugin, PrecisionPlugin
2626
from pytorch_lightning.plugins.training_type import DataParallelPlugin, TrainingTypePlugin
2727
from pytorch_lightning.trainer.states import TrainerFn
28+
from pytorch_lightning.utilities import rank_zero_deprecation
2829
from pytorch_lightning.utilities.apply_func import apply_to_collection, move_data_to_device
2930
from pytorch_lightning.utilities.enums import AMPType, LightningEnum
3031
from pytorch_lightning.utilities.types import STEP_OUTPUT
@@ -56,6 +57,15 @@ def __init__(
5657
self.training_type_plugin = training_type_plugin
5758

5859
if precision_plugin:
60+
"""
61+
.. deprecated
62+
precision_plugin parameter is deprecated will be removed soon.
63+
Use :`training_type_plugin(precision_plugin) instead.
64+
"""
65+
rank_zero_deprecation(
66+
f"`{self.__class__.__name__}.precision` was and will be removed soon"
67+
f" Use `training_type_plugin.precision_plugin.precision` instead."
68+
)
5969
self.training_type_plugin._precision_plugin = precision_plugin
6070

6171
self.optimizers: List = []
@@ -213,7 +223,7 @@ def optimizer_step(
213223
opt_idx: int,
214224
closure: Callable[[], Any],
215225
model: Optional[Union["pl.LightningModule", Module]] = None,
216-
**kwargs: Any
226+
**kwargs: Any,
217227
) -> None:
218228
"""performs the actual optimizer step.
219229
@@ -270,8 +280,16 @@ def amp_backend(self) -> Optional[LightningEnum]:
270280

271281
@property
272282
def precision(self) -> Union[str, int]:
273-
"""deprecated."""
274-
return self.training_type_plugin.precision
283+
"""
284+
.. deprecated
285+
This method is deprecated will be removed soon.
286+
Use :`training_type_plugin.precision_plugin.precision` instead.
287+
"""
288+
rank_zero_deprecation(
289+
f"`{self.__class__.__name__}.precision` was and will be removed soon"
290+
f" Use `training_type_plugin.precision_plugin.precision` instead."
291+
)
292+
return self.training_type_plugin.precision_plugin.precision
275293

276294
@property
277295
def scaler(self) -> Optional["GradScaler"]:

pytorch_lightning/plugins/training_type/training_type_plugin.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,10 +47,6 @@ def __init__(
4747
def checkpoint_io(self) -> CheckpointIO:
4848
return self._checkpoint_io
4949

50-
@property
51-
def precision(self) -> Union[str, int]:
52-
return self._precision_plugin.precision
53-
5450
@property
5551
def precision_plugin(self) -> PrecisionPlugin:
5652
return self._precision_plugin

0 commit comments

Comments
 (0)