update

projects/mock_transformers/mock_tokenization.py

@@ -16,12 +16,12 @@
 import os
 
 import oneflow as flow
+import oneflow.mock_torch as mock
 
 from libai.utils import distributed as dist
 import oneflow.mock_torch as mock
 
 with mock.enable(lazy=True):
-
     from transformers import (  # noqa
         BertTokenizer,
         GPT2Tokenizer,
@@ -33,6 +33,7 @@
     from transformers.utils import generic  # noqa
     from transformers.utils.generic import TensorType  # noqa
 
+
     # ---------------- mock TensorType ------------------
     class TensorType(ExplicitEnum):  # noqa
         PYTORCH = "pt"
@@ -77,6 +78,7 @@ def flow_convert_to_tensors(self, tensor_type=None, prepend_batch_axis=False):
             try:
                 import oneflow  # noqa
             except ImportError as e:
+
                 msg = (
                     "Unable to convert output to OneFlow tensors format, OneFlow is not installed."
                 )
@@ -143,4 +145,5 @@ def flow_convert_to_tensors(self, tensor_type=None, prepend_batch_axis=False):
                 self[k] = v.to_global(sbp=sbp, placement=dist.get_layer_placement(0))
         return self
 
+
     BatchEncoding.convert_to_tensors = flow_convert_to_tensors  # noqa