Fix imports in tests/gflownet/envs/test_lattice_parameters.py

alexhernandezgarcia · Mar 4, 2024 · 2fbb792 · 2fbb792
1 parent fc4c535
commit 2fbb792
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/gflownet/envs/test_lattice_parameters.py b/tests/gflownet/envs/test_lattice_parameters.py
@@ -9,19 +9,18 @@
 import pytest
 import torch
 
-from gflownet.envs.crystals.lattice_parameters import (
+from gflownet.envs.crystals.lattice_parameters import PARAMETER_NAMES, LatticeParameters
+from gflownet.utils.common import tfloat
+from gflownet.utils.crystals.constants import (
     CUBIC,
     HEXAGONAL,
     LATTICE_SYSTEMS,
     MONOCLINIC,
     ORTHORHOMBIC,
-    PARAMETER_NAMES,
     RHOMBOHEDRAL,
     TETRAGONAL,
     TRICLINIC,
-    LatticeParameters,
 )
-from gflownet.utils.common import tfloat
 
 N_REPETITIONS = 100