Corrected bug in ContextualModule

torch_modules/src/ContextualModule.cpp

@@ -173,11 +173,15 @@ void ContextualModuleImpl::addToContext(std::vector<std::vector<long>> & context
   for (auto index : targetIndexes)
   {
     if (configIndex2ContextIndex.count(index))
+    {
       for (auto & contextElement : context)
-        contextElement.push_back(configIndex2ContextIndex.at(index)+1);
+        contextElement.push_back(configIndex2ContextIndex.at(index));
+    }
     else
+    {
       for (auto & contextElement : context)
         contextElement.push_back(0);
+    }
   }
 }