Snakemake: как реализовать функцию с использованием подстановочных знаков? ⇐ Python

[Anonymous]

Я пытаюсь использовать Snakemake для вывода некоторых файлов из определенного задания.
По сути, у меня есть разные каналы процесса, которые охватывают разные диапазоны масс. В зависимости от пары {канал, масса} мне приходится запускать задания для разных значений «норм». Я хотел сделать это, используя:

Код: Выделить всё

import numpy as np
import pickle as pkl

particle_masses = {
5: 4.18,     # Bottom quark mass ~4.18 GeV however there is an issue with the charon spectra below 10 GeV
8: 80.3,   # W boson mass ~80.379 GeV
11: 1.777,   # Tau lepton mass ~1.777 GeV
12: 0,       # Electron neutrino mass ~0 (neutrino masses are very small)
14: 0,       # Muon neutrino mass ~0
16: 0        # Tau neutrino mass ~0
}

mass_values = np.logspace(0, np.log10(499), num=25).tolist()
# Masked mass dictionary with rounded values
masked_mass_dict = {}
for channel, min_mass in particle_masses.items():
# Apply mask to exclude values below the particle mass threshold
masked_mass = [round(m, 2) for m in mass_values if m >= max(min_mass, 3)]
masked_mass_dict[channel] = masked_mass

channels = list(masked_mass_dict.keys())

rule all:
input:
expand(
f"{DATA_LOC}/signal/channel_{{channel}}/trial_distributions/{{mass}}_trial_distrib_{{norm}}.h5",
channel=channels,
mass=lambda wildcards: masked_mass_dict[wildcards.channel],
norm=lambda wildcards: get_norms({"mass": wildcards.mass, "channel": wildcards.channel}),
)

rule compute_trial_distribution:
input:
signal_file=f"{DATA_LOC}/signal/channel_{{channel}}/mc_distrib/{{mass}}_mc_distrib.h5",

output:
norm_file=f"{DATA_LOC}/signal/channel_{{channel}}/trial_distributions/{{mass}}_trial_distrib_{{norm}}.h5"
shell:
"""
...
"""

def get_norms(dict):
"""
Get the norm values (sensitivity) for the given channel and mass from the pre- loaded sensitivity dictionary.
"""

channel = dict["channel"]
mass = dict["mass"]
channel = int(channel)
mass = float(mass)
#channel = int(wildcards.channel)
#mass = float(wildcards.mass)

# Load sensitivity dictionary
dictionary_file = "path/"
with open(dictionary_file, "rb") as f:
sensitivities_dict = pkl.load(f)

# Get the sensitivity data for the specified channel
if channel not in sensitivities_dict:
raise ValueError(f"Channel {channel} not found in sensitivity dictionary.")

sensitivity_mass_array = sensitivities_dict[channel]
mass_index = np.where(sensitivity_mass_array[0] == mass)[0]

if len(mass_index) == 0:
raise ValueError(f"Mass {mass} not found for channel {channel}")

# Calculate norms
azimov_norm = sensitivity_mass_array[1][mass_index[0]]
norms = np.linspace(0.1 * azimov_norm, 10 * azimov_norm, 50)
norms = np.insert(norms, 0, 0)  # Include a norm value of 0 for background-only distribution
norms = np.array([0])
# Convert to list for use in Snakemake params or shell
return norms.tolist()

Однако похоже, что это не работает. Я не уверен, как это правильно реализовать... Вот такая ошибка:

Код: Выделить всё

InputFunctionException in rule all in file /home/egenton/upgrade_solar_WIMP/scripts/Snakefile, line 45:
Error:
AttributeError: 'Wildcards' object has no attribute 'channel'
Wildcards:

Если кто-нибудь знает, как это можно решить, это было бы очень полезно!
Я попробовал ввести пары масс каналов, чтобы расширить нормы из get_norm функция, но это не сработало, поскольку функция getnorm распознает подстановочный знак.

Подробнее здесь: https://stackoverflow.com/questions/792 ... -wildcards