From 831ad565c1d57a790419002cbb1bd84d6f234c63 Mon Sep 17 00:00:00 2001
From: Carel van Niekerk <40663106+carelvniekerk@users.noreply.github.com>
Date: Thu, 15 Dec 2022 11:11:54 +0100
Subject: [PATCH] Update vector_base.py
---
convlab/policy/vector/vector_base.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/convlab/policy/vector/vector_base.py b/convlab/policy/vector/vector_base.py
index a9e096e4..d85cca10 100644
--- a/convlab/policy/vector/vector_base.py
+++ b/convlab/policy/vector/vector_base.py
@@ -316,7 +316,7 @@ class VectorBase(Vector):
entities list:
list of entities of the specified domain
"""
- constraints = self.state[domain]
+ constraints = self.state.get(domain, {})
# Leave slots out of constraints to find which slot constraint results in no entities being found
for constraint_slot in constraints:
--
GitLab