Skip to content

Commit

Permalink
Fix: Addressed 'max' test failures in pull request
Browse files Browse the repository at this point in the history
  • Loading branch information
m-momeni authored and m-momeni committed Dec 13, 2024
1 parent b072861 commit eb2f730
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions tests/test_modeling_flax_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -1148,9 +1148,6 @@ def test_gradient_checkpointing(self):
for output, remat_output in zip(outputs, remat_outputs):
self.assertTrue((output == remat_output).all())

@require_torch
@require_flax
@is_pt_flax_cross_test
def test_activation_fns(self):
# Assuming Torch activation functions of `activations.ACT2FN` as a base, compares Flax implementations
# to produce equal/close results.
Expand Down

0 comments on commit eb2f730

Please sign in to comment.