diff --git a/analogvnn/graph/AcyclicDirectedGraph.py b/analogvnn/graph/AcyclicDirectedGraph.py
index 7f14b19..f333918 100644
--- a/analogvnn/graph/AcyclicDirectedGraph.py
+++ b/analogvnn/graph/AcyclicDirectedGraph.py
@@ -131,8 +131,8 @@ def add_edge(
         self.graph.nodes[v_of_edge]['fillcolor'] = 'lightblue'
         return self
 
-    @staticmethod
-    def check_edge_parameters(  # noqa: C901
+    @staticmethod  # noqa: C901
+    def check_edge_parameters(
             in_arg: Union[None, int, bool],
             in_kwarg: Union[None, str, bool],
             out_arg: Union[None, int, bool],
diff --git a/analogvnn/nn/module/Layer.py b/analogvnn/nn/module/Layer.py
index 6a3ca65..f73630e 100644
--- a/analogvnn/nn/module/Layer.py
+++ b/analogvnn/nn/module/Layer.py
@@ -204,8 +204,7 @@ def named_registered_children(
             self,
             memo: Optional[Set[nn.Module]] = None
     ) -> Iterator[Tuple[str, nn.Module]]:
-        """Returns an iterator over immediate registered children modules, yielding both
-        the name of the module.
+        """Returns an iterator over immediate registered children modules.
 
         Args:
             memo: a memo to store the set of modules already added to the result
diff --git a/analogvnn/nn/module/Sequential.py b/analogvnn/nn/module/Sequential.py
index c16d49b..e06afb5 100644
--- a/analogvnn/nn/module/Sequential.py
+++ b/analogvnn/nn/module/Sequential.py
@@ -51,5 +51,5 @@ def add_sequence(self, *args):
         Args:
             *args (nn.Module): The modules to add.
         """
-        
+
         return self.extend(args)
diff --git a/pyproject.toml b/pyproject.toml
index 0f24fc3..51a3cb0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -102,7 +102,7 @@ flake8 = [
     "flake8-executable",
     "flake8-coding",
     "flake8-return",
-    "flake8-noreturn",
+#    "flake8-noreturn; python_version >= '3.8'",
     "flake8-deprecated",
 ]
 dev = [
diff --git a/requirements.txt b/requirements.txt
index d5668b2..464c65c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,12 +5,12 @@ torchaudio
 numpy
 scipy
 networkx
-importlib-metadata<5.0.0,>=2.0.0; python_version < "3.8"
+importlib-metadata<5.0.0,>=2.0.0; python_version < '3.8'
 
 # Full
-tensorflow
-tensorboard
+tensorflow>=2.0.0
+tensorboard>=2.0.0
 torchinfo
 pillow
-# conda install python-graphviz graphviz
+# conda install graphviz
 graphviz
diff --git a/requirements/requirements-test.txt b/requirements/requirements-test.txt
index f67ad55..54c409a 100644
--- a/requirements/requirements-test.txt
+++ b/requirements/requirements-test.txt
@@ -7,5 +7,5 @@ flake8-comprehensions
 flake8-executable
 flake8-coding
 flake8-return
-flake8-noreturn
+# flake8-noreturn>=1.0.1; python_version >= '3.8'
 flake8-deprecated