added annotations

Degiacomi-Lab · Aug 6, 2024 · 05a399b · 05a399b
1 parent ad6baec
commit 05a399b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/molearn/models/transformer.py b/src/molearn/models/transformer.py
@@ -1,3 +1,4 @@
+from __future__ import annotations
 import torch
 import torch.nn as nn
 

diff --git a/src/molearn/trainers/transformer_trainer.py b/src/molearn/trainers/transformer_trainer.py
@@ -1,3 +1,4 @@
+from __future__ import annotations
 import os
 from .trainer import Trainer