missing import

GFNOrg · Dec 12, 2024 · b133740 · b133740
1 parent 9bbe332
commit b133740
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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