Instantly share code, notes, and snippets.

View gist:1b656a376fd471122a62f0d9eda99ebd
#!/bin/bash
#SBATCH --job-name=notebook
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=8
#SBATCH --mem-per-cpu=4480
#SBATCH --time=12:00:00
#SBATCH --account=pi-lgrandi
#SBATCH --partition=dali
View load_data
#!/bin/bash
########################
### Datasets loading ###
########################
import datetime
dsets_version = 0.0
import numpy as np
import sys
import hax
View gist:361e3b4727120095f3b726c1f011267d
from hax.treemakers.peak_treemakers import PeakExtractor
class CoinPeaksInTPC(PeakExtractor):
__version__ = '0.0.3'
stop_after = 10
peak_fields = ['area', 'type', 'x', 'y', 'n_hits', 'area_per_channel']
event_cut_list = ['event_number %2 == 0' ]
peak_cut_list = ['detector == "tpc"', 'type !="lone_hit"']
extra_branches = ['peaks.area', 'peaks.type', 'peaks.reconstructed_positions*', 'peaks.n_hits', 'peaks.detector']
d_tpit = hax.minitrees.load(dataset, [CoinPeaksInTPC, 'Fundamentals'])