From 0735ceba02b4fccb61b6f0ab8fff81a50b12d284 Mon Sep 17 00:00:00 2001 From: Joseph Date: Thu, 23 Nov 2023 11:56:51 -0500 Subject: [PATCH] isort --- src/gfn/gym/helpers/box_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/gfn/gym/helpers/box_utils.py b/src/gfn/gym/helpers/box_utils.py index dd9acd01..c6342c75 100644 --- 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