From 3035fde3cabebc88873e6395f03b4fb101ca138d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Edgar=20Du=C3=A9=C3=B1ez-Guzm=C3=A1n?= <duenez@google.com>
Date: Sun, 7 Jul 2024 09:51:24 -0700
Subject: [PATCH] Add missing superclass `overrides.EnforceOverrides`

PiperOrigin-RevId: 650025315
Change-Id: Id9f352031b9f907a6e8e7bd6b66e3b9e4124830f
---
 concordia/typing/component_v2.py | 3 ++-
 concordia/typing/memory.py       | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/concordia/typing/component_v2.py b/concordia/typing/component_v2.py
index b98abff..7781ef8 100644
--- 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
index f5f5bbf..81de365 100644
--- 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