Remove inheritance from object

examples/fairseq/tasks/data/utils.py

@@ -62,7 +62,7 @@ class NativeCheckpointableIterator(iterators.CheckpointableIterator):
         pass
 
 
-class WeightIterator(object):
+class WeightIterator:
     def __init__(self, weights, seed):
         self.weights = weights
         self.seed = seed

torchscale/architecture/config.py

@@ -2,7 +2,7 @@
 # Licensed under The MIT License [see LICENSE for details]
 
 
-class EncoderConfig(object):
+class EncoderConfig:
     def __init__(self, **kwargs):
         self.encoder_embed_dim = kwargs.pop("encoder_embed_dim", 768)
         self.encoder_attention_heads = kwargs.pop("encoder_attention_heads", 12)
@@ -71,7 +71,7 @@ class EncoderConfig(object):
                 self.__dict__[hp] = getattr(args, hp, None)
 
 
-class DecoderConfig(object):
+class DecoderConfig:
     def __init__(self, **kwargs):
         self.decoder_embed_dim = kwargs.pop("decoder_embed_dim", 768)
         self.decoder_attention_heads = kwargs.pop("decoder_attention_heads", 12)
@@ -135,7 +135,7 @@ class DecoderConfig(object):
                 self.__dict__[hp] = getattr(args, hp, None)
 
 
-class EncoderDecoderConfig(object):
+class EncoderDecoderConfig:
     def __init__(self, **kwargs):
         self.encoder_embed_dim = kwargs.pop("encoder_embed_dim", 768)
         self.encoder_attention_heads = kwargs.pop("encoder_attention_heads", 12)

torchscale/component/feedforward_network.py

@@ -13,7 +13,7 @@ except ModuleNotFoundError:
 from .xmoe.global_groups import get_moe_group
 
 
-class set_torch_seed(object):
+class set_torch_seed:
     def __init__(self, seed):
         assert isinstance(seed, int)
         self.rng_state = self.get_rng_state()