working model

Reconstruction/VKalVrt/GNNVertexConstructor/CMakeLists.txt

@@ -91,6 +91,7 @@ atlas_add_library(   GNNVertexConstructorLib
 # Install files from the package:
 atlas_install_python_modules( python/*.py POST_BUILD_CMD ${ATLAS_FLAKE8} )
 
+atlas_install_data( data/* )
 
 
 

Reconstruction/VKalVrt/GNNVertexConstructor/Root/GNNVertexConstructorTool.cxx

@@ -23,7 +23,7 @@ GNNVertexConstructorTool::GNNVertexConstructorTool(const std::string &type, cons
 
   declareProperty("JetTrackLinks", m_trackLinksKey = "BTagging_AntiKt4EMPFlowAuxDyn.TrackLinks");
   declareProperty("JetVertexLinks",
-                  m_vertexLinksKey = "BTagging_AntiKt4EMPFlowAuxDyn.track_vertexing");
+                  m_vertexLinksKey = "BTagging_AntiKt4EMPFlowAuxDyn.vertex_indices");
 
   declareProperty("GNNTool", m_gnn_Tool, "GNN Tool");
   declareProperty("VertexFitterTool", m_vertexFitterTool, "Vertex fitting tool");

Reconstruction/VKalVrt/GNNVertexConstructor/python/GNNVertexConstructorToolConfig.py

@@ -12,7 +12,7 @@ def GNNVertexConstructorToolCfg(flags, name="LMEdevTool", **kwargs):
     gnnTool = acc.getPrimaryAndMerge(
             GNNToolCfg(
                 flags,
-                NNFile           ="../network.onnx",
+                NNFile           ="GNNVertexConstructor/network.onnx",
                 trackLinkType    ="IPARTICLE",  #Either IPARTICLE or  TRACK_PARTICLE
                 variableRemapping={"BTagTrackToJetAssociator" : "GhostTrack"},
                 )