From b133740b0b9c36b5a5f55005eed7719ebb051950 Mon Sep 17 00:00:00 2001
From: Joseph Viviano <joseph@viviano.ca>
Date: Thu, 12 Dec 2024 11:06:43 -0500
Subject: [PATCH] missing import

---
 src/gfn/utils/training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/gfn/utils/training.py b/src/gfn/utils/training.py
index 4e6359e..8206bc2 100644
--- a/src/gfn/utils/training.py
+++ b/src/gfn/utils/training.py
@@ -1,5 +1,5 @@
 from collections import Counter
-from typing import Dict, Optional
+from typing import Dict, Optional, Tuple
 
 import torch
 from torchtyping import TensorType as TT