AFT-738: Disable GNN vertex fitter, again

AFT-738: Disable GNN vertex fitter, again

PhysicsAnalysis/DerivationFramework/DerivationFrameworkFlavourTag/python/FtagDerivationConfig.py

@@ -11,7 +11,6 @@ from FlavorTagDiscriminants.FoldDecoratorConfig import FoldDecoratorCfg
 from ElectronPhotonSelectorTools.AsgElectronLikelihoodToolsConfig import AsgElectronLikelihoodToolCfg
 from ElectronPhotonSelectorTools.LikelihoodEnums import LikeEnum
 from MuonSelectorTools.MuonSelectorToolsConfig import MuonSelectionToolCfg
-from AthenaConfiguration.Enums import LHCPeriod
 
 PFLOW_JETS = 'AntiKt4EMPFlowJets'
 
@@ -66,7 +65,7 @@ def FtagJetCollectionsCfg(cfgFlags, jet_cols, pv_cols=None,
             )
         )
     
-    if cfgFlags.GeoModel.Run < LHCPeriod.Run4:
+    if cfgFlags.BTagging.GNNVertexFitter:
       from GNNVertexFitter.GNNVertexFitterConfig import GNNVertexFitterAlgCfg
       acc.merge(GNNVertexFitterAlgCfg(cfgFlags, name="GNNVertexFitterAlg"))
     

PhysicsAnalysis/JetTagging/JetTagConfig/python/BTaggingConfigFlags.py

@@ -142,6 +142,9 @@ def createBTaggingConfigFlags():
     # a flag to add V0finder
     btagcf.addFlag("BTagging.AddV0Finder", False)
 
+    # GNN vertex fitter
+    btagcf.addFlag("BTagging.GNNVertexFitter", False)
+
     # (multifold) NN trainings, each jet collection maps to a list of
     # dicts. The dict has several keys:
     #  - folds: list of NNs to run