diff --git a/hannah/callbacks/optimization.py b/hannah/callbacks/optimization.py
index 57784b92..7f7fc2a2 100644
--- a/hannah/callbacks/optimization.py
+++ b/hannah/callbacks/optimization.py
@@ -94,7 +94,7 @@ def _add_monitor_mapping(self, monitor):
         else:
             self.directions.append(-1.0)
 
-    def on_train_batch_end(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule",  outputs: 'STEP_OUTPUT', batch: Any, batch_idx: int) -> None:
+    def on_train_batch_end(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule",  outputs: 'STEP_OUTPUT', batch: Any, batch_idx: int) -> None:  # noqa: F821
         callback_metrics =  trainer.callback_metrics
 
         for k, v in callback_metrics.items():
diff --git a/hannah/nas/spaces/mobilenet/mobilenet.py b/hannah/nas/spaces/mobilenet/mobilenet.py
index f5eaa31d..91093a54 100644
--- a/hannah/nas/spaces/mobilenet/mobilenet.py
+++ b/hannah/nas/spaces/mobilenet/mobilenet.py
@@ -13,7 +13,7 @@ def residual(input):
 
 
 @dataflow
-def add(input, other):
+def add(input, other): # noqa: F811
     return op("Add", input, other)