Fix minor issue

NKI-AI · Sep 20, 2023 · b680111 · b680111
1 parent f1b8c83
commit b680111
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/direct/nn/mri_models.py b/direct/nn/mri_models.py
@@ -2,6 +2,7 @@
 # Copyright (c) DIRECT Contributors
 
 """MRI model engine of DIRECT."""
+from __future__ import annotations
 
 import gc
 import pathlib