version 1.5.0

TylerYep · Jul 3, 2021 · 49498c1 · 49498c1
1 parent f28fa62
commit 49498c1
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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