Simplify state conversions in Composition and missing bit in CCrystal.

alexhernandezgarcia · Dec 9, 2023 · b07e258 · b07e258
1 parent 261b8fb
commit b07e258
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gflownet/envs/crystals/ccrystal.py b/gflownet/envs/crystals/ccrystal.py
@@ -903,7 +903,7 @@ def states2proxy(
         states = tfloat(states, device=self.device, float_type=self.float)
         return torch.cat(
             [
-                subenv.states2oracle(self._get_states_of_subenv(states, stage))
+                subenv.states2proxy(self._get_states_of_subenv(states, stage))
                 for stage, subenv in self.subenvs.items()
             ],
             dim=1,