From 1ced84a06f6999403f93945795164a0bb9a9113b Mon Sep 17 00:00:00 2001 From: Geoffrey Yu Date: Wed, 29 Nov 2023 15:17:08 -0500 Subject: [PATCH] Fix watchdog check --- src/brad/daemon/blueprint_watchdog.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/brad/daemon/blueprint_watchdog.py b/src/brad/daemon/blueprint_watchdog.py index ca965831..46920164 100644 --- a/src/brad/daemon/blueprint_watchdog.py +++ b/src/brad/daemon/blueprint_watchdog.py @@ -33,7 +33,7 @@ def reject_blueprint(self, blueprint: Blueprint) -> bool: # Embedding table should not leave Aurora. try: embedding_locations = blueprint.get_table_locations("embeddings") - if embedding_locations != [Engine.Aurora]: + if Engine.Aurora not in embedding_locations: if self._event_logger is not None: self._event_logger.log( SystemEvent.WatchdogFired,