From 49498c1845872bb29840cda7abb9ea5187b1a9f2 Mon Sep 17 00:00:00 2001 From: Tyler Yep Date: Fri, 2 Jul 2021 17:49:56 -0700 Subject: [PATCH] version 1.5.0 --- fixtures/models.py | 2 +- setup.cfg | 2 +- tests/depth_test.py | 2 +- torchinfo/layer_info.py | 2 +- torchinfo/torchinfo.py | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/fixtures/models.py b/fixtures/models.py index 996fb42..5224bf8 100644 --- a/fixtures/models.py +++ b/fixtures/models.py @@ -4,7 +4,7 @@ from typing import Any, Dict, Tuple, cast import torch -import torch.nn as nn +from torch import nn from torch.nn import functional as F from torch.nn.utils.rnn import pack_padded_sequence diff --git a/setup.cfg b/setup.cfg index de8714c..2c8f1bb 100644 --- a/setup.cfg +++ b/setup.cfg @@ -1,6 +1,6 @@ [metadata] name = torchinfo -version = 0.1.5 +version = 1.5.0 description = Model summary in PyTorch, based off of the original torchsummary. long_description = file: README.md long_description_content_type = text/markdown diff --git a/tests/depth_test.py b/tests/depth_test.py index 9b2b0d7..8afe9ee 100644 --- a/tests/depth_test.py +++ b/tests/depth_test.py @@ -3,7 +3,7 @@ from collections import namedtuple import torch -import torch.nn as nn +from torch import nn from fixtures.models import IdentityModel from torchinfo import summary diff --git a/torchinfo/layer_info.py b/torchinfo/layer_info.py index fce23e8..3c3a5dd 100644 --- a/torchinfo/layer_info.py +++ b/torchinfo/layer_info.py @@ -2,7 +2,7 @@ from typing import Any, Dict, Iterable, List, Optional, Sequence, Union import torch -import torch.nn as nn +from torch import nn from torch.jit import ScriptModule DETECTED_INPUT_OUTPUT_TYPES = Union[ diff --git a/torchinfo/torchinfo.py b/torchinfo/torchinfo.py index 0834cf5..3662b65 100644 --- a/torchinfo/torchinfo.py +++ b/torchinfo/torchinfo.py @@ -14,7 +14,7 @@ ) import torch -import torch.nn as nn +from torch import nn from torch.jit import ScriptModule from torch.utils.hooks import RemovableHandle