Refactor FLAX tests (#9034)

2020-12-10 15:57:39 -05:00
parent 1310e1a758
commit 8d4bb02056
4 changed files with 294 additions and 110 deletions
--- a/src/transformers/tokenization_utils_base.py
+++ b/src/transformers/tokenization_utils_base.py
@@ -50,6 +50,7 @@ if is_tf_available():

 if is_torch_available():
    import torch
+
 if is_flax_available():
    import jax.numpy as jnp