Add optional custom log_rewards to all GFlowNet losses - #503

Merged

josephdviviano merged 11 commits into

custom-log-rewards

Apr 3, 2026

Commits on Apr 3, 2026

Add optional custom log_rewards to all GFlowNet losses

josephdviviano
and
claude
committed
Add end-to-end tests for custom log_rewards parameter

josephdviviano
and
claude
committed
Initial plan
Copilot
authored
fix: accept log_rewards kwarg in TBGAFN.get_scores with default None

Copilot
and
josephdviviano
authored
Merge pull request #504 from GFNOrg/copilot/fix-ci-failure-log-rewards
josephdviviano
authored
Update src/gfn/gflownet/sub_trajectory_balance.py

josephdviviano
and
Copilot
authored
fix: make log_rewards keyword-only in all loss() and get_scores() methods for backward compatibility

Copilot
and
josephdviviano
authored
Update testing/test_custom_log_rewards.py

josephdviviano
and
Copilot
authored
feat: raise ValueError when forward_looking=True with custom log_rewards; add TrustPCL and RelativeLPV tests

Copilot
and
josephdviviano
authored
Fix black formatting for CI (unstable string concatenation rules)

josephdviviano
and
claude
committed
black
josephdviviano
committed