Skip to content

Commit 21a2137

Browse files
committed
delay import
Signed-off-by: xavier dupré <[email protected]>
1 parent 5393d59 commit 21a2137

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_string_ops.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,6 @@
1111
import unittest
1212
import numpy as np
1313
import tensorflow as tf
14-
from tensorflow.python.ops.ragged import ragged_tensor
15-
from tensorflow.python.ops import lookup_ops
1614

1715
from backend_test_base import Tf2OnnxBackendTestBase
1816
from common import requires_custom_ops
@@ -150,6 +148,8 @@ def run_onnxruntime(self, model_path, inputs, output_names):
150148
def test_wordpiece_tokenizer(self):
151149
from tensorflow_text.python.ops.wordpiece_tokenizer import (
152150
gen_wordpiece_tokenizer as lib_gen_wordpiece_tokenizer)
151+
from tensorflow.python.ops.ragged import ragged_tensor
152+
from tensorflow.python.ops import lookup_ops
153153

154154
def _CreateTable(vocab, num_oov=1):
155155
init = tf.lookup.KeyValueTensorInitializer(

0 commit comments

Comments
 (0)