add annotations for compatibility with older python versions

module.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import torch
 from torch import nn
 from torch.nn import BCEWithLogitsLoss