Adapt common tests to new GFlowNetAgent env_maker parameter

alexhernandezgarcia · Mar 19, 2024 · 9d9a3d7 · 9d9a3d7
1 parent 92f4d83
commit 9d9a3d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/gflownet/envs/common.py b/tests/gflownet/envs/common.py
@@ -477,7 +477,7 @@ def test__gflownet_minimal_runs(self, n_repeat=1):
                 config.gflownet,
                 device=config.device,
                 float_precision=config.float_precision,
-                env=self.env,
+                env_maker=self.env.__class__,
                 forward_policy=forward_policy,
                 backward_policy=backward_policy,
                 buffer=config.env.buffer,