Merge pull request #515 from BindsNET/hananel

Minor syntax corrections
BindsNET · Sep 10, 2021 · 85e44b7 · 85e44b7
2 parents 37b4c09 + 8a76f40
commit 85e44b7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/bindsnet/conversion/conversion.py b/bindsnet/conversion/conversion.py
@@ -133,8 +133,8 @@ def set_requires_grad(module, value):
 
                         prev_factor = scale_factor
 
-                elif isinstance(module2, nn.Linear) or isinstance(module2, nn.Conv2d):
-                    prev_module = module2
+                elif isinstance(module, nn.Linear) or isinstance(module, nn.Conv2d):
+                    prev_module = module
 
         if isinstance(module, nn.Linear):
             if prev_module is not None:

diff --git a/bindsnet/network/monitors.py b/bindsnet/network/monitors.py
@@ -8,6 +8,11 @@
 from .nodes import Nodes
 from .topology import AbstractConnection
 
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .network import Network
+
 
 class AbstractMonitor(ABC):
     # language=rst