Decorate config() as final - i.e., overriding it in a subclass is not…

… permitted PiperOrigin-RevId: 559538483
google · Aug 23, 2023 · 49a17a2 · 49a17a2
1 parent b89ade0
commit 49a17a2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/praxis/base_layer.py b/praxis/base_layer.py
@@ -1428,6 +1428,7 @@ def __to_sub_config_field__(cls):
   def __call__(cls, *args, **kwargs):
     return pax_fiddle.Config(cls.fiddle_base_layer_cls, *args, **kwargs)
 
+  @typing.final
   def config(cls, **kwargs):
     return cls(**kwargs)
 
@@ -2489,6 +2490,7 @@ def _quantize_fn(self, return_pspec: bool) -> NestedJTensor | Any:
         res[target][var_name] = var_val
     return res
 
+  @typing.final
   @classmethod
   def config(cls, **kwargs) -> pax_fiddle.Config:
     return pax_fiddle.Config(cls, **kwargs)