Skip to content

Commit 0890b9b

Browse files
committed
Merge commit 'refs/pull/20728/head' of https://github.com/Lightning-AI/pytorch-lightning into dependabot-pip-requirements-torch-2.6.0
2 parents 64132fb + 44314cd commit 0890b9b

File tree

1 file changed

+3
-2
lines changed
  • src/lightning/pytorch/plugins/precision

1 file changed

+3
-2
lines changed

src/lightning/pytorch/plugins/precision/double.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,9 @@
1111
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
14-
from collections.abc import Generator
14+
from collections.abc import Generator, Iterable
1515
from contextlib import AbstractContextManager, contextmanager
16-
from typing import Any, Literal, Iterable
16+
from typing import Any, Literal
1717

1818
import torch
1919
import torch.nn as nn
@@ -71,6 +71,7 @@ class LightningDoublePrecisionModule(_DeviceDtypeModuleMixin, nn.Module):
7171
pl_module: the model to wrap
7272
7373
"""
74+
7475
_ddp_params_and_buffers_to_ignore: Iterable[str]
7576

7677
def __init__(self, pl_module: "pl.LightningModule") -> None:

0 commit comments

Comments
 (0)