moscot.base.problems.BirthDeathProblem.set_solution

BirthDeathProblem.set_solution(solution, *, overwrite=False, **kwargs)

Set a solution to the OT problem.

Parameters:

• solution (Union[ndarray[Any, dtype[float64]], DataFrame, BaseSolverOutput]) – Solution for this problem. If a DataFrame is passed, its index must be equal to adata_src.obs_names and its columns to adata_tgt.obs_names.

• overwrite (bool) – Whether to overwrite an existing solution.

• kwargs (Any) – Keyword arguments for MatrixSolverOutput.

Return type:

OTProblem

Returns:

: Self and updates the following fields:

• solution - the OT solution.

• stage - set to 'solved'.