From 2aed8dbb8db0b370f1ce680c9919194d32f03a6b Mon Sep 17 00:00:00 2001 From: awaelchli Date: Mon, 5 Aug 2024 01:31:59 +0200 Subject: [PATCH] update --- src/lightning/fabric/utilities/distributed.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lightning/fabric/utilities/distributed.py b/src/lightning/fabric/utilities/distributed.py index 87c00b022d105..0e6c52dfb09b9 100644 --- a/src/lightning/fabric/utilities/distributed.py +++ b/src/lightning/fabric/utilities/distributed.py @@ -7,14 +7,14 @@ from contextlib import nullcontext from datetime import timedelta from pathlib import Path -from typing import TYPE_CHECKING, Any, Iterable, Iterator, List, Optional, Sized, TypeGuard, Union +from typing import TYPE_CHECKING, Any, Iterable, Iterator, List, Optional, Sized, Union import torch import torch.nn.functional as F from lightning_utilities.core.imports import package_available from torch import Tensor from torch.utils.data import Dataset, DistributedSampler, Sampler -from typing_extensions import Self, override +from typing_extensions import Self, TypeGuard, override from lightning.fabric.utilities.cloud_io import _is_local_file_protocol from lightning.fabric.utilities.data import _num_cpus_available