From 59ddb0d280d59b502c02a7ce623b8743f92618c9 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Mon, 29 Jun 2020 10:28:09 +0200
Subject: [PATCH] Adding None to dict in cotnextual module
---
torch_modules/src/ContextualModule.cpp | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/torch_modules/src/ContextualModule.cpp b/torch_modules/src/ContextualModule.cpp
index d5f6078..d857a2d 100644
--- a/torch_modules/src/ContextualModule.cpp
+++ b/torch_modules/src/ContextualModule.cpp
@@ -132,7 +132,12 @@ void ContextualModuleImpl::addToContext(std::vector<std::vector<long>> & context
else if (index == -2)
{
for (auto & contextElement : context)
+ {
+ auto currentState = dict.getState();
+ dict.setState(Dict::State::Open);
contextElement.push_back(dict.getIndexOrInsert(fmt::format("{}({})", col, "_NONE_")));
+ dict.setState(currentState);
+ }
}
else
{
--
GitLab