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