bug fix

alexhernandezgarcia · Nov 27, 2023 · 1d7a838 · 1d7a838
1 parent 9f4dea2
commit 1d7a838
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gflownet/gflownet.py b/gflownet/gflownet.py
@@ -199,10 +199,10 @@ def parameters(self):
             parameters += list(self.backward_policy.model.parameters())
         if self.state_flow is not None:
             if self.loss != "forwardlooking":
-                raise ValueError(f"State flow cannot be trained in {self.loss} loss.")
+                raise ValueError(f"State flow cannot be trained with {self.loss} loss.")
             parameters += list(self.state_flow.model.parameters())
         return parameters
-                raise ValueError(f"State flow cannot be trained with {self.loss} loss.")
+
     def sample_actions(
         self,
         envs: List[GFlowNetEnv],