When adding new arguments to a class it's important not to change the order of existing arguments, just in case someone is calling them positionally. You should add weight_constraints as the last argument, after `init_bias_target_marginals`.

It probably doesn't matter in this case but it's a good habit to develop.