From 05a399b2743166cf44a027502464035a3136675d Mon Sep 17 00:00:00 2001
From: gwirn <71886945+gwirn@users.noreply.github.com>
Date: Tue, 6 Aug 2024 12:15:14 +0200
Subject: [PATCH] added annotations

---
 src/molearn/models/transformer.py           | 1 +
 src/molearn/trainers/transformer_trainer.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/molearn/models/transformer.py b/src/molearn/models/transformer.py
index 9f5966f..1982092 100644
--- 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
index 47f2fda..4ef1e7d 100644
--- 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