Skip to content

Commit

Permalink
Minor fix
Browse files Browse the repository at this point in the history
  • Loading branch information
fchollet committed Dec 3, 2024
1 parent 58ac150 commit e072189
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -143,7 +143,7 @@ def _equalize_channel(self, channel, value_range):
def _apply_equalization(self, channel, hist):
cdf = self.backend.numpy.cumsum(hist)

if backend.backend() == "jax":
if self.backend.name == "jax":
mask = cdf > 0
first_nonzero_idx = self.backend.numpy.argmax(mask)
cdf_min = self.backend.numpy.take(cdf, first_nonzero_idx)
Expand Down

0 comments on commit e072189

Please sign in to comment.