diff --git a/Tools/PyJobTransforms/python/trfAMI.py b/Tools/PyJobTransforms/python/trfAMI.py index 2c3fd4044963d5f9d076bc8c6f361ece88fda70a..8cbf661b2c92eff49689215f27f3495fc5028754 100644 --- a/Tools/PyJobTransforms/python/trfAMI.py +++ b/Tools/PyJobTransforms/python/trfAMI.py @@ -356,13 +356,11 @@ def getTrfConfigFromPANDA(tag): if arg.lstrip('-').startswith('input') and arg.endswith('File'): value=physics.pop(arg) msg.debug("Found input file argument %s=%s.", arg, value ) - fmt=arg.lstrip('-').replace('input','').replace('File','') - trf.inFiles[arg]=getInputFileName(arg) + trf.inFiles[arg]=value elif arg.lstrip('-').startswith('output') and arg.endswith('File'): value=physics.pop(arg) msg.debug("Found output file argument %s=%s.", arg, value ) - fmt=arg.lstrip('-').replace('output','').replace('File','') - trf.outFiles[arg]=getOutputFileName(fmt) + trf.outFiles[arg]=value msg.debug("Checking for not set arguments...") for arg,value in listitems(physics): @@ -471,7 +469,7 @@ def getTrfConfigFromAMI(tag, suppressNonJobOptions = True): if suppressNonJobOptions: for k in list(physics): - if k in ['productionStep', 'transformation', 'SWReleaseCache']: + if k in ['inputs', 'outputs', 'productionStep', 'transformation', 'SWReleaseCache']: physics.pop(k) for k, v in iteritems(physics): @@ -490,13 +488,11 @@ def getTrfConfigFromAMI(tag, suppressNonJobOptions = True): if arg.lstrip('-').startswith('input') and arg.endswith('File'): value = physics.pop(arg) msg.debug("Found input file argument %s=%s.", arg, value) - fmt = arg.lstrip('-').replace('input', '').replace('File', '') - trf.inFiles[arg] = getInputFileName(arg) + trf.inFiles[arg] = value elif arg.lstrip('-').startswith('output') and arg.endswith('File'): value = physics.pop(arg) msg.debug("Found output file argument %s=%s.", arg, value) - fmt = arg.lstrip('-').replace('output', '').replace('File', '') - trf.outFiles[arg] = getOutputFileName(fmt) + trf.outFiles[arg] = value msg.debug("Checking for not set arguments...") for arg, value in listitems(physics): diff --git a/Tools/PyJobTransforms/test/test_trfAMI.py b/Tools/PyJobTransforms/test/test_trfAMI.py index 143d070c90ce1edfa9e4f3f0aa734489bbfccbc2..7a53f02421ff81992f58fd43be7b87dcac82461e 100755 --- a/Tools/PyJobTransforms/test/test_trfAMI.py +++ b/Tools/PyJobTransforms/test/test_trfAMI.py @@ -154,7 +154,9 @@ class trfAMIUnitTests(unittest.TestCase): self.assertEqual(tag.trfs[0].release, 'AtlasProduction,19.1.1.9') self.assertEqual(tag.trfs[0].newTransform, True) self.assertEqual(tag.trfs[0].physics, physics) - self.assertEqual(tag.trfs[0].inFiles, {'inputHighPtMinbiasHitsFile': 'myHighPtMinbiasHits', 'inputLowPtMinbiasHitsFile': 'myLowPtMinbiasHits'}) + self.assertEqual(tag.trfs[0].inFiles, { + 'inputHighPtMinbiasHitsFile': 'mc14_13TeV.119996.Pythia8_A2MSTW2008LO_minbias_inelastic_high.merge.HITS.e3038_s1967_s2007', + 'inputLowPtMinbiasHitsFile': 'mc14_13TeV.119995.Pythia8_A2MSTW2008LO_minbias_inelastic_low.merge.HITS.e3038_s1967_s2007'}) self.assertEqual(tag.trfs[0].outFiles, {}) self.assertEqual(tag.trfs[0].outputs, {}) self.assertEqual(tag.trfs[0].inDS, None) @@ -223,8 +225,9 @@ class trfAMIUnitTests(unittest.TestCase): self.assertEqual(tag.trfs[0].release, 'AtlasProduction,20.1.4.5') self.assertEqual(tag.trfs[0].newTransform, True) self.assertEqual(tag.trfs[0].physics, physics) - self.assertEqual(tag.trfs[0].inFiles, {'inputHighPtMinbiasHitsFile': 'myHighPtMinbiasHits', - 'inputLowPtMinbiasHitsFile': 'myLowPtMinbiasHits'}) + self.assertEqual(tag.trfs[0].inFiles, { + 'inputHighPtMinbiasHitsFile': 'mc15_valid.361035.Pythia8EvtGen_A2MSTW2008LO_minbias_inelastic_high.merge.HITS.e3581_s2578_s2169_tid05098387_00', + 'inputLowPtMinbiasHitsFile': 'mc15_valid.361034.Pythia8EvtGen_A2MSTW2008LO_minbias_inelastic_low.merge.HITS.e3581_s2578_s2169_tid05098374_00'}) self.assertEqual(tag.trfs[0].outFiles, {}) self.assertEqual(tag.trfs[0].outputs, {}) self.assertEqual(tag.trfs[0].inDS, None)