Refactor: Removed un-necessary object base class (#32230)
* Refactored to remove un-necessary object base class. * small fix.
This commit is contained in:
@@ -37,7 +37,7 @@ if is_tf_available():
|
||||
)
|
||||
|
||||
|
||||
class TFCTRLModelTester(object):
|
||||
class TFCTRLModelTester:
|
||||
def __init__(
|
||||
self,
|
||||
parent,
|
||||
|
||||
@@ -34,7 +34,7 @@ if is_torch_available():
|
||||
)
|
||||
|
||||
|
||||
class DebertaModelTester(object):
|
||||
class DebertaModelTester:
|
||||
def __init__(
|
||||
self,
|
||||
parent,
|
||||
|
||||
@@ -35,7 +35,7 @@ if is_torch_available():
|
||||
)
|
||||
|
||||
|
||||
class DebertaV2ModelTester(object):
|
||||
class DebertaV2ModelTester:
|
||||
def __init__(
|
||||
self,
|
||||
parent,
|
||||
|
||||
@@ -40,7 +40,7 @@ if is_torch_available():
|
||||
from transformers.models.distilbert.modeling_distilbert import _create_sinusoidal_embeddings
|
||||
|
||||
|
||||
class DistilBertModelTester(object):
|
||||
class DistilBertModelTester:
|
||||
def __init__(
|
||||
self,
|
||||
parent,
|
||||
|
||||
@@ -39,7 +39,7 @@ if is_torch_available():
|
||||
from transformers.models.flaubert.modeling_flaubert import create_sinusoidal_embeddings
|
||||
|
||||
|
||||
class FlaubertModelTester(object):
|
||||
class FlaubertModelTester:
|
||||
def __init__(
|
||||
self,
|
||||
parent,
|
||||
|
||||
@@ -34,7 +34,7 @@ if is_tf_available():
|
||||
from transformers.models.lxmert.modeling_tf_lxmert import TFLxmertForPreTraining, TFLxmertModel
|
||||
|
||||
|
||||
class TFLxmertModelTester(object):
|
||||
class TFLxmertModelTester:
|
||||
def __init__(
|
||||
self,
|
||||
parent,
|
||||
|
||||
@@ -84,7 +84,7 @@ class TFMobileBertModelTest(TFModelTesterMixin, PipelineTesterMixin, unittest.Te
|
||||
|
||||
return inputs_dict
|
||||
|
||||
class TFMobileBertModelTester(object):
|
||||
class TFMobileBertModelTester:
|
||||
def __init__(
|
||||
self,
|
||||
parent,
|
||||
|
||||
@@ -37,7 +37,7 @@ if is_torch_available():
|
||||
)
|
||||
|
||||
|
||||
class SqueezeBertModelTester(object):
|
||||
class SqueezeBertModelTester:
|
||||
def __init__(
|
||||
self,
|
||||
parent,
|
||||
|
||||
@@ -23,7 +23,7 @@ from transformers import is_torch_available
|
||||
from .utils.test_configuration_utils import config_common_kwargs
|
||||
|
||||
|
||||
class ConfigTester(object):
|
||||
class ConfigTester:
|
||||
def __init__(self, parent, config_class=None, has_text_modality=True, common_properties=None, **kwargs):
|
||||
self.parent = parent
|
||||
self.config_class = config_class
|
||||
|
||||
Reference in New Issue
Block a user