Skip to content

Commit 03aba35

Browse files
fix tools_maker
1 parent b325800 commit 03aba35

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

sbndcode/WireCell/cfg/pgrapher/experiment/sbnd/wcls-nf-sp.jsonnet

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,14 +33,15 @@ local dnnroi_model_p1 = std.extVar('dnnroi_model_p1');
3333
local g = import 'pgraph.jsonnet';
3434
local f = import 'pgrapher/experiment/sbnd/funcs.jsonnet';
3535
local wc = import 'wirecell.jsonnet';
36-
local default_tools = tools_maker(params);
37-
local tools = if wc_device == 'gpu' then std.mergePatch(default_tools,
38-
{dft: {type: "TorchDFT", data: {device: wc_device}}}) else default_tools;
36+
local tools_maker = import 'pgrapher/common/tools.jsonnet';
3937

4038
local simu_params = import 'simparams.jsonnet';
4139
local params = simu_params;
4240

43-
local tools = tools_maker(params);
41+
local default_tools = tools_maker(params);
42+
local tools = if wc_device == 'gpu' then std.mergePatch(default_tools,
43+
{dft: {type: "TorchDFT", data: {device: wc_device}}}) else default_tools;
44+
4445

4546
local mega_anode = {
4647
type: 'MegaAnodePlane',

0 commit comments

Comments
 (0)