Replace methodName with *args **kwargs (#3359)
diff --git a/integrations/tensorflow/e2e/README.md b/integrations/tensorflow/e2e/README.md
index 553faf7..c5a8f8f 100644
--- a/integrations/tensorflow/e2e/README.md
+++ b/integrations/tensorflow/e2e/README.md
@@ -83,8 +83,8 @@
# Inherit from `TracedModuleTestCase`.
class SimpleArithmeticTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(SimpleArithmeticTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(SimpleArithmeticTest, self).__init__(*args, **kwargs)
# Compile a `tf.Module` named `SimpleArithmeticModule` into
# `CompiledModule`s for each reference and target backend.
self._modules = tf_test_utils.compile_tf_module(SimpleArithmeticModule)
diff --git a/integrations/tensorflow/e2e/batch_norm_test.py b/integrations/tensorflow/e2e/batch_norm_test.py
index 4e3c0cd..da49161 100644
--- a/integrations/tensorflow/e2e/batch_norm_test.py
+++ b/integrations/tensorflow/e2e/batch_norm_test.py
@@ -42,8 +42,8 @@
class BatchNormTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(BatchNormTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(BatchNormTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(BatchNormModule)
def test_batch_norm_inference(self):
diff --git a/integrations/tensorflow/e2e/bool_test.py b/integrations/tensorflow/e2e/bool_test.py
index ef9dd10..e086cff 100644
--- a/integrations/tensorflow/e2e/bool_test.py
+++ b/integrations/tensorflow/e2e/bool_test.py
@@ -40,8 +40,8 @@
class BooleanTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(BooleanTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(BooleanTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(MathModule)
def test_constant(self):
diff --git a/integrations/tensorflow/e2e/broadcast_to_test.py b/integrations/tensorflow/e2e/broadcast_to_test.py
index dc53f34..4f6ec87 100644
--- a/integrations/tensorflow/e2e/broadcast_to_test.py
+++ b/integrations/tensorflow/e2e/broadcast_to_test.py
@@ -33,8 +33,8 @@
class BroadcastToTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(BroadcastToTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(BroadcastToTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(BroadcastToModule)
def test_scalar_broadcast_to(self):
diff --git a/integrations/tensorflow/e2e/broadcasting_test.py b/integrations/tensorflow/e2e/broadcasting_test.py
index f72f3b3..7a4750e 100644
--- a/integrations/tensorflow/e2e/broadcasting_test.py
+++ b/integrations/tensorflow/e2e/broadcasting_test.py
@@ -33,8 +33,8 @@
class BroadcastingTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(BroadcastingTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(BroadcastingTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(BroadcastingModule)
def test_add_same_shape(self):
diff --git a/integrations/tensorflow/e2e/complex_test.py b/integrations/tensorflow/e2e/complex_test.py
index 4832146..2a42bcd 100644
--- a/integrations/tensorflow/e2e/complex_test.py
+++ b/integrations/tensorflow/e2e/complex_test.py
@@ -35,8 +35,8 @@
class ComplexTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(ComplexTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(ComplexTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(ComplexModule)
def test_complex(self):
diff --git a/integrations/tensorflow/e2e/concat_test.py b/integrations/tensorflow/e2e/concat_test.py
index dc0fab8..f7747e6 100644
--- a/integrations/tensorflow/e2e/concat_test.py
+++ b/integrations/tensorflow/e2e/concat_test.py
@@ -54,8 +54,8 @@
class ConcatOpsTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(ConcatOpsTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(ConcatOpsTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(ConcatOpsModule)
def test_concat_zero_dim(self):
diff --git a/integrations/tensorflow/e2e/control_flow_test.py b/integrations/tensorflow/e2e/control_flow_test.py
index 4ba691a..790c6c7 100644
--- a/integrations/tensorflow/e2e/control_flow_test.py
+++ b/integrations/tensorflow/e2e/control_flow_test.py
@@ -37,8 +37,8 @@
class ControlFlowTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(ControlFlowTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(ControlFlowTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(ControlFlowModule)
def test_short_sequence(self):
diff --git a/integrations/tensorflow/e2e/conv_test.py b/integrations/tensorflow/e2e/conv_test.py
index cff4496..f1a6a94 100644
--- a/integrations/tensorflow/e2e/conv_test.py
+++ b/integrations/tensorflow/e2e/conv_test.py
@@ -102,8 +102,8 @@
class ConvTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(ConvTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(ConvTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(Conv2dModule)
def test_id_batch_size_1(self):
diff --git a/integrations/tensorflow/e2e/depth_conv_test.py b/integrations/tensorflow/e2e/depth_conv_test.py
index c928d5e..db80010 100644
--- a/integrations/tensorflow/e2e/depth_conv_test.py
+++ b/integrations/tensorflow/e2e/depth_conv_test.py
@@ -76,8 +76,8 @@
class ConvTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(ConvTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(ConvTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(DepthConv2dModule)
def test_batched_feature_unpadded(self):
diff --git a/integrations/tensorflow/e2e/dynamic_mlp_relu_test.py b/integrations/tensorflow/e2e/dynamic_mlp_relu_test.py
index 1e5ba08..511b770 100644
--- a/integrations/tensorflow/e2e/dynamic_mlp_relu_test.py
+++ b/integrations/tensorflow/e2e/dynamic_mlp_relu_test.py
@@ -67,8 +67,8 @@
class DynamicMlpReluTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(DynamicMlpReluTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(DynamicMlpReluTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(DynamicMlpReluModule,
exported_names=["predict"])
diff --git a/integrations/tensorflow/e2e/dynamic_mlp_test.py b/integrations/tensorflow/e2e/dynamic_mlp_test.py
index a0ecdc5..2675014 100644
--- a/integrations/tensorflow/e2e/dynamic_mlp_test.py
+++ b/integrations/tensorflow/e2e/dynamic_mlp_test.py
@@ -63,8 +63,8 @@
class DynamicMlpTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(DynamicMlpTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(DynamicMlpTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(DynamicMlpModule,
exported_names=["predict"])
diff --git a/integrations/tensorflow/e2e/fill_test.py b/integrations/tensorflow/e2e/fill_test.py
index adaeac9..d622074 100644
--- a/integrations/tensorflow/e2e/fill_test.py
+++ b/integrations/tensorflow/e2e/fill_test.py
@@ -33,8 +33,8 @@
class FillTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(FillTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(FillTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(FillModule)
def test_fill(self):
diff --git a/integrations/tensorflow/e2e/finite_test.py b/integrations/tensorflow/e2e/finite_test.py
index 1f53406..fb5f5ca 100644
--- a/integrations/tensorflow/e2e/finite_test.py
+++ b/integrations/tensorflow/e2e/finite_test.py
@@ -27,8 +27,8 @@
class FiniteTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(FiniteTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(FiniteTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(FiniteModule)
def test_finite(self):
diff --git a/integrations/tensorflow/e2e/gather_test.py b/integrations/tensorflow/e2e/gather_test.py
index da71d4b..cf81c9b 100644
--- a/integrations/tensorflow/e2e/gather_test.py
+++ b/integrations/tensorflow/e2e/gather_test.py
@@ -68,8 +68,8 @@
class GatherTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(GatherTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(GatherTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(GatherModule)
def test_gather_axis0_scalar(self):
diff --git a/integrations/tensorflow/e2e/keras/lstm_static_test.py b/integrations/tensorflow/e2e/keras/lstm_static_test.py
index 5cb1827..96cd6b3 100644
--- a/integrations/tensorflow/e2e/keras/lstm_static_test.py
+++ b/integrations/tensorflow/e2e/keras/lstm_static_test.py
@@ -45,8 +45,8 @@
class LstmStaticTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(LstmStaticTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(LstmStaticTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(LstmStaticModule,
exported_names=["predict"])
diff --git a/integrations/tensorflow/e2e/keras/lstm_test.py b/integrations/tensorflow/e2e/keras/lstm_test.py
index 747cc6f..f831673 100644
--- a/integrations/tensorflow/e2e/keras/lstm_test.py
+++ b/integrations/tensorflow/e2e/keras/lstm_test.py
@@ -41,8 +41,8 @@
class LstmTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(LstmTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(LstmTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(LstmModule,
exported_names=["predict"])
diff --git a/integrations/tensorflow/e2e/keras/vision_model_test.py b/integrations/tensorflow/e2e/keras/vision_model_test.py
index 528b8c4..0e7a263 100644
--- a/integrations/tensorflow/e2e/keras/vision_model_test.py
+++ b/integrations/tensorflow/e2e/keras/vision_model_test.py
@@ -145,8 +145,8 @@
class AppTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(AppTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(AppTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(VisionModule,
exported_names=['predict'])
diff --git a/integrations/tensorflow/e2e/linspace_test.py b/integrations/tensorflow/e2e/linspace_test.py
index 4acc170..46072a2 100644
--- a/integrations/tensorflow/e2e/linspace_test.py
+++ b/integrations/tensorflow/e2e/linspace_test.py
@@ -36,8 +36,8 @@
class LinspaceTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(LinspaceTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(LinspaceTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(LinspaceModule)
def test_linspace(self):
diff --git a/integrations/tensorflow/e2e/logical_ops_test.py b/integrations/tensorflow/e2e/logical_ops_test.py
index ee83a95..ae4f328 100644
--- a/integrations/tensorflow/e2e/logical_ops_test.py
+++ b/integrations/tensorflow/e2e/logical_ops_test.py
@@ -49,8 +49,8 @@
class LogicalOpsTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(LogicalOpsTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(LogicalOpsTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(LogicalOpsModule)
def test_logical_and(self):
diff --git a/integrations/tensorflow/e2e/mandelbrot_test.py b/integrations/tensorflow/e2e/mandelbrot_test.py
index 51c1509..9f2a881 100644
--- a/integrations/tensorflow/e2e/mandelbrot_test.py
+++ b/integrations/tensorflow/e2e/mandelbrot_test.py
@@ -93,8 +93,8 @@
class MandelbrotTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(MandelbrotTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(MandelbrotTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(MandelbrotModule)
def test_mandelbrot(self):
diff --git a/integrations/tensorflow/e2e/math_test.py b/integrations/tensorflow/e2e/math_test.py
index a96193a..178e1a9 100644
--- a/integrations/tensorflow/e2e/math_test.py
+++ b/integrations/tensorflow/e2e/math_test.py
@@ -45,8 +45,8 @@
class MathTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(MathTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(MathTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(MathModule)
def test_abs(self):
diff --git a/integrations/tensorflow/e2e/matrix_ops_dynamic_test.py b/integrations/tensorflow/e2e/matrix_ops_dynamic_test.py
index 4a3daf7..21224da 100644
--- a/integrations/tensorflow/e2e/matrix_ops_dynamic_test.py
+++ b/integrations/tensorflow/e2e/matrix_ops_dynamic_test.py
@@ -46,8 +46,8 @@
class MatrixOpsDynamicTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(MatrixOpsDynamicTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(MatrixOpsDynamicTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(MatrixOpsDynamicModule)
def test_matmul_high_rank_batch(self):
diff --git a/integrations/tensorflow/e2e/matrix_ops_static_test.py b/integrations/tensorflow/e2e/matrix_ops_static_test.py
index 3a7fe47..c49230a 100644
--- a/integrations/tensorflow/e2e/matrix_ops_static_test.py
+++ b/integrations/tensorflow/e2e/matrix_ops_static_test.py
@@ -58,8 +58,8 @@
class MatrixOpsStaticTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(MatrixOpsStaticTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(MatrixOpsStaticTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(MatrixOpsStaticModule)
def test_basic_matmul(self):
diff --git a/integrations/tensorflow/e2e/mobile_bert_squad_test.py b/integrations/tensorflow/e2e/mobile_bert_squad_test.py
index b99759d..e674a2b 100644
--- a/integrations/tensorflow/e2e/mobile_bert_squad_test.py
+++ b/integrations/tensorflow/e2e/mobile_bert_squad_test.py
@@ -44,8 +44,8 @@
class MobileBertSquadTest(tf_test_utils.TracedModuleTestCase):
"""Tests of MobileBertSquad."""
- def __init__(self, methodName='runTest'):
- super(MobileBertSquadTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(MobileBertSquadTest, self).__init__(*args, **kwargs)
model_type = 'quant_saved_model' if FLAGS.use_quantized_weights else 'float'
# Get_file will download the model weights from a publicly available folder,
diff --git a/integrations/tensorflow/e2e/range_test.py b/integrations/tensorflow/e2e/range_test.py
index f4e3e97..a7673a8 100644
--- a/integrations/tensorflow/e2e/range_test.py
+++ b/integrations/tensorflow/e2e/range_test.py
@@ -34,8 +34,8 @@
class RangeTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(RangeTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(RangeTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(RangeModule)
def test_range(self):
diff --git a/integrations/tensorflow/e2e/resource_ops_test.py b/integrations/tensorflow/e2e/resource_ops_test.py
index d387e86..668ef37 100644
--- a/integrations/tensorflow/e2e/resource_ops_test.py
+++ b/integrations/tensorflow/e2e/resource_ops_test.py
@@ -31,8 +31,8 @@
class ResourcesOpsTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(ResourcesOpsTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(ResourcesOpsTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(ResourcesOpsModule)
def test_add_assign(self):
diff --git a/integrations/tensorflow/e2e/ring_buffer_test.py b/integrations/tensorflow/e2e/ring_buffer_test.py
index c56bf31..a3c8d18 100644
--- a/integrations/tensorflow/e2e/ring_buffer_test.py
+++ b/integrations/tensorflow/e2e/ring_buffer_test.py
@@ -185,8 +185,8 @@
class StatefulRingBufferTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(StatefulRingBufferTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(StatefulRingBufferTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(StatefulRingBufferModule,
exported_names=["predict"])
diff --git a/integrations/tensorflow/e2e/scatter_update_test.py b/integrations/tensorflow/e2e/scatter_update_test.py
index 8f34002..ca7528d 100644
--- a/integrations/tensorflow/e2e/scatter_update_test.py
+++ b/integrations/tensorflow/e2e/scatter_update_test.py
@@ -51,8 +51,8 @@
class ScatterUpdateTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(ScatterUpdateTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(ScatterUpdateTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(ScatterUpdateModule)
def test_scatter_update_1D(self):
diff --git a/integrations/tensorflow/e2e/simple_arithmetic_test.py b/integrations/tensorflow/e2e/simple_arithmetic_test.py
index cc9ca09..314338b 100644
--- a/integrations/tensorflow/e2e/simple_arithmetic_test.py
+++ b/integrations/tensorflow/e2e/simple_arithmetic_test.py
@@ -40,8 +40,8 @@
class SimpleArithmeticTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(SimpleArithmeticTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(SimpleArithmeticTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(SimpleArithmeticModule)
def test_simple_mul(self):
diff --git a/integrations/tensorflow/e2e/simple_stateful_test.py b/integrations/tensorflow/e2e/simple_stateful_test.py
index e4140d4..788b4c2 100644
--- a/integrations/tensorflow/e2e/simple_stateful_test.py
+++ b/integrations/tensorflow/e2e/simple_stateful_test.py
@@ -36,8 +36,8 @@
class StatefulTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(StatefulTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(StatefulTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(SimpleStatefulModule)
def test_stateful(self):
diff --git a/integrations/tensorflow/e2e/sliding_window_test.py b/integrations/tensorflow/e2e/sliding_window_test.py
index 413ffb2..5450f6e 100644
--- a/integrations/tensorflow/e2e/sliding_window_test.py
+++ b/integrations/tensorflow/e2e/sliding_window_test.py
@@ -78,8 +78,8 @@
class SlidingWindowTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(SlidingWindowTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(SlidingWindowTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(SlidingWindowModule,
exported_names=["predict"])
diff --git a/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py b/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py
index 7439f77..735885a 100644
--- a/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py
+++ b/integrations/tensorflow/e2e/slim_vision_models/slim_vision_model_test.py
@@ -75,8 +75,8 @@
class SlimVisionTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(SlimVisionTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(SlimVisionTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(SlimVisionModule,
exported_names=['predict'])
diff --git a/integrations/tensorflow/e2e/strings_test.py b/integrations/tensorflow/e2e/strings_test.py
index 70f6468..0d1bca1 100644
--- a/integrations/tensorflow/e2e/strings_test.py
+++ b/integrations/tensorflow/e2e/strings_test.py
@@ -43,8 +43,8 @@
class StringsTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(StringsTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(StringsTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(StringsModule)
def test_print_ids(self):
diff --git a/integrations/tensorflow/e2e/tensorlist_test.py b/integrations/tensorflow/e2e/tensorlist_test.py
index 62babc5..00f3259 100644
--- a/integrations/tensorflow/e2e/tensorlist_test.py
+++ b/integrations/tensorflow/e2e/tensorlist_test.py
@@ -70,8 +70,8 @@
class TensorListTest(tf_test_utils.TracedModuleTestCase):
- def __init__(self, methodName="runTest"):
- super(TensorListTest, self).__init__(methodName)
+ def __init__(self, *args, **kwargs):
+ super(TensorListTest, self).__init__(*args, **kwargs)
self._modules = tf_test_utils.compile_tf_module(TensorListModule)
def test_identity_through_tensorlist(self):
