From 83283fdb208e0f8a3388453e2abd62527cfc493f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Carlos=20Mochol=C3=AD?= Date: Wed, 12 May 2021 15:44:57 +0200 Subject: [PATCH] Fix yapf-isort conflict (#7500) --- pytorch_lightning/utilities/types.py | 22 ++++++++++++++++++---- 1 file changed, 18 insertions(+), 4 deletions(-) diff --git a/pytorch_lightning/utilities/types.py b/pytorch_lightning/utilities/types.py index ecb0101a22..fdfdb95b08 100644 --- a/pytorch_lightning/utilities/types.py +++ b/pytorch_lightning/utilities/types.py @@ -1,12 +1,26 @@ -from typing import Any, Dict, Iterator, List, Union - -import torch -from torchmetrics import Metric +# Copyright The PyTorch Lightning team. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. """ Convention: - Do not include any `_TYPE` suffix - Types used in public hooks (as those in the `LightningModule` and `Callback`) should be public (no trailing `_`) """ +from typing import Any, Dict, Iterator, List, Union + +import torch +from torchmetrics import Metric + _METRIC = Union[Metric, torch.Tensor, int, float] STEP_OUTPUT = Union[torch.Tensor, Dict[str, Any]] EPOCH_OUTPUT = List[STEP_OUTPUT]