Exemplo n.º 1
0
def main(args):
    jobs = JDLCreator("docker")

    jobs.executable = "gof/job.sh"
    jobs.wall_time = 5 * 60 * 60
    jobs.memory = 4096
    jobs.accounting_group = "cms.higgs"
    jobs.image = "mschnepf/slc7-condocker"

    # Build list of arguments
    arguments = []
    binning = yaml.load(open(args.binning))
    variables = yaml.load(open(args.variables))

    # 1D fits
    for channel in binning["gof"]:
        for variable in variables["selected_variables"][int(args.era)][channel]:
            arguments.append("{} {} {} {}".format(args.era, channel, variable, os.getcwd()))

    # 2D fits
    for channel in binning["gof"]:
        selected_2d_fits = []
        for var1 in variables["selected_variables"][int(args.era)][channel]:
            for var2 in variables["selected_variables"][int(args.era)][channel]:
                selected_2d_fits.append("{}_{}".format(var1, var2))
        for variable in selected_2d_fits:
            if variable in binning["gof"][channel]:
                arguments.append("{} {} {} {}".format(args.era, channel, variable, os.getcwd()))

    jobs.arguments = arguments

    # The job requires lots of CPU resources
    jobs.requirements = '(Target.ProvidesCPU == True) && (Target.ProvidesIO == True ) && (Target.ProvidesEKPResources == True)'
    jobs.job_folder = args.output
    jobs.WriteJDL()
Exemplo n.º 2
0
def main(args):
    jobs = JDLCreator("ekpsupermachines")

    jobs.executable = "ml/job.sh"
    jobs.wall_time = 10 * 60 * 60
    jobs.memory = 2048
    jobs.accounting_group = "cms.higgs"
    jobs.image = "stwunsch/slc6-condocker:smhtt"

    # Build list of arguments
    arguments = []
    for modifier in args.modifiers:
        arguments.append("{} {}".format(args.channel, modifier))
    jobs.arguments = arguments

    # The job requires lots of CPU resources
    # NOTE: This selects the sg machines.
    jobs.requirements = "(Target.ProvidesCPU == True) && (Target.ProvidesEKPResources == True)"
    jobs.job_folder = args.output
    jobs.WriteJDL()
Exemplo n.º 3
0
def main(args):
    jobs = JDLCreator("ekpsupermachines")

    jobs.executable = "gof/job.sh"
    jobs.wall_time = 1 * 60 * 60
    jobs.memory = 2048
    jobs.accounting_group = "cms.higgs"
    jobs.image = "stwunsch/slc6-condocker:smhtt"

    # Build list of arguments
    arguments = []
    config = yaml.load(open(args.binning))
    for channel in config["gof"]:
        for variable in config["gof"][channel]:
            arguments.append("{} {}".format(channel, variable))
    jobs.arguments = arguments

    # The job requires lots of CPU resources
    # NOTE: This selects the sg machines.
    jobs.requirements = '(Target.ProvidesCPU == True) && (Target.ProvidesEKPResources == True)'
    jobs.job_folder = args.output
    jobs.WriteJDL()