Corrected bug in ContextualModule

torch_modules/src/ContextualModule.cpp

@@ -173,13 +173,17 @@ 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);
     }
   }
+}
 
 torch::Tensor batchedIndexSelect(torch::Tensor input, int dim, torch::Tensor index)
 {