Add missing superclass overrides.EnforceOverrides

PiperOrigin-RevId: 650025315 Change-Id: Id9f352031b9f907a6e8e7bd6b66e3b9e4124830f
google-deepmind · Jul 7, 2024 · 3035fde · 3035fde
1 parent fe939f9
commit 3035fde
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/concordia/typing/component_v2.py b/concordia/typing/component_v2.py
@@ -21,6 +21,7 @@
 from typing import Any
 
 from concordia.typing import entity as entity_lib
+import overrides
 
 ComponentName = str
 ComponentContext = str
@@ -75,7 +76,7 @@ def get_component(self, component_name: str) -> "BaseComponent":
     raise NotImplementedError()
 
 
-class BaseComponent:
+class BaseComponent(overrides.EnforceOverrides):
   """A base class for components."""
 
   def __init__(self):

diff --git a/concordia/typing/memory.py b/concordia/typing/memory.py
@@ -17,6 +17,7 @@
 import abc
 from collections.abc import Mapping, Sequence
 from typing import Any, Protocol
+import overrides
 
 
 class MemoryScorer(Protocol):
@@ -32,7 +33,7 @@ def __call__(self, query: str, text: str, **metadata: Any) -> float:
     """
 
 
-class MemoryBank(metaclass=abc.ABCMeta):
+class MemoryBank(overrides.EnforceOverrides, metaclass=abc.ABCMeta):
   """Base class for memory banks."""
 
   @abc.abstractmethod