isort

GFNOrg · Nov 23, 2023 · 0735ceb · 0735ceb
1 parent f4f69d6
commit 0735ceb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/gfn/gym/helpers/box_utils.py b/src/gfn/gym/helpers/box_utils.py
@@ -1,11 +1,11 @@
 """This file contains utilitary functions for the Box environment."""
 from typing import Tuple
 
-from torch.distributions import Beta, Categorical, Distribution, MixtureSameFamily
-from torchtyping import TensorType as TT
 import numpy as np
 import torch
 import torch.nn as nn
+from torch.distributions import Beta, Categorical, Distribution, MixtureSameFamily
+from torchtyping import TensorType as TT
 
 from gfn.gym import Box
 from gfn.modules import GFNModule