moscot.base.problems.CondOTProblem.prepare

CondOTProblem.prepare(policy_key, policy, xy, xx, conditions, a=None, b=None, subset=None, reference=None, **kwargs)[source]

Prepare conditional optimal transport problem.

Parameters:
Return type:

CondOTProblem

Returns:

: Self and modifies the following attributes: TODO.