From c841324271af889717cf050ce82fa1f82b7c327f Mon Sep 17 00:00:00 2001
From: theissenhelen <helen.theissen@ecmwf.int>
Date: Thu, 3 Oct 2024 09:45:10 +0000
Subject: [PATCH] fix: import annotations from future

---
 src/anemoi/models/layers/attention.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/anemoi/models/layers/attention.py b/src/anemoi/models/layers/attention.py
index e7e0db2..6c333dd 100644
--- a/src/anemoi/models/layers/attention.py
+++ b/src/anemoi/models/layers/attention.py
@@ -7,6 +7,8 @@
 # nor does it submit to any jurisdiction.
 #
 
+from __future__ import annotations
+
 import logging
 import math
 from typing import Optional