DepthLayerTreeEmbedding no longer contains gov

torch_modules/src/DepthLayerTreeEmbedding.cpp

@@ -70,13 +70,7 @@ void DepthLayerTreeEmbeddingImpl::addToContext(std::vector<std::vector<long>> &
             newChilds.insert(newChilds.end(), val.begin(), val.end());
           }
         childs = newChilds;
-        if (depth == 0)
-        {
-          newChilds.clear();
-          auto gov = config.has(0,index,0) ? config.getAsFeature(Config::headColName, index).get() : "-1";
-          newChilds.emplace_back(util::isEmpty(gov) ? "-1" : gov);
-          newChilds.insert(newChilds.end(), childs.begin(), childs.end());
-        }
+
         for (int i = 0; i < maxElemPerDepth[depth]; i++)
           for (auto & col : columns)
             if (i < (int)newChilds.size() and config.has(col, std::stoi(newChilds[i]), 0))

torch_modules/src/LSTMNetwork.cpp

@@ -26,7 +26,7 @@ LSTMNetworkImpl::LSTMNetworkImpl(int nbOutputs, int unknownValueThreshold, std::
   if (!treeEmbeddingColumns.empty())
   {
     hasTreeEmbedding = true;
-    treeEmbedding = register_module("treeEmbedding", DepthLayerTreeEmbedding(treeEmbeddingNbElems,embeddingsSize,treeEmbeddingSize,treeEmbeddingColumns,treeEmbeddingBuffer,treeEmbeddingStack,lstmOptionsAll));
+    treeEmbedding = register_module("treeEmbedding", DepthLayerTreeEmbedding(treeEmbeddingNbElems,embeddingsSize,treeEmbeddingSize,treeEmbeddingColumns,treeEmbeddingBuffer,treeEmbeddingStack,lstmOptions));
     treeEmbedding->setFirstInputIndex(currentInputSize);
     currentOutputSize += treeEmbedding->getOutputSize();
     currentInputSize += treeEmbedding->getInputSize();