Fix watchdog check

mitdbg · Nov 29, 2023 · 1ced84a · 1ced84a
1 parent 0da91fd
commit 1ced84a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,