Skip to content

Commit

Permalink
updated names
Browse files Browse the repository at this point in the history
  • Loading branch information
Intron7 committed Dec 19, 2024
1 parent a051cf9 commit 610b15e
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/rapids_singlecell/preprocessing/_harmony/_fuses.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,8 @@ def _get_pen(E: cp.ndarray, O: cp.ndarray, theta: cp.ndarray) -> cp.ndarray:


@cp.fuse
def _calc_R(term: cp.ndarray, mm: cp.ndarray) -> cp.ndarray:
return cp.exp(term * (1 - mm))
def _calc_R(term: cp.ndarray, dotproduct: cp.ndarray) -> cp.ndarray:
return cp.exp(term * (1 - dotproduct))


@cp.fuse
Expand All @@ -24,5 +24,5 @@ def _log_div_OE(O: cp.ndarray, E: cp.ndarray) -> cp.ndarray:


@cp.fuse
def _R_multi_m(R: cp.ndarray, other: cp.ndarray) -> cp.ndarray:
return R * 2 * (1 - other)
def _R_multi_m(R: cp.ndarray, dotproduct: cp.ndarray) -> cp.ndarray:
return R * 2 * (1 - dotproduct)

0 comments on commit 610b15e

Please sign in to comment.