From 11dba025368b7377045192ad43b84c8f8a18932f Mon Sep 17 00:00:00 2001
From: Maxim Gonchar <maxim.mg.gonchar@gmail.com>
Date: Wed, 29 Mar 2023 11:18:50 +0300
Subject: [PATCH] feat: add Graph to dayabay_v0
---
models/dayabay_v0.py | 15 +++++++++++----
1 file changed, 11 insertions(+), 4 deletions(-)
diff --git a/models/dayabay_v0.py b/models/dayabay_v0.py
index 2a000d5..d7191f0 100644
--- a/models/dayabay_v0.py
+++ b/models/dayabay_v0.py
@@ -5,6 +5,9 @@ from pathlib import Path
from typing import Union, Tuple, List, Optional
from pandas import DataFrame
+from dagflow.graph import Graph
+from dagflow.graphviz import savegraph
+
class ParametersWrapper(NestedMKDict):
def to_dict(self, **kwargs) -> list:
data = []
@@ -43,10 +46,12 @@ class ParametersWrapper(NestedMKDict):
def model_dayabay_v0():
storage = ParametersWrapper({}, sep='.')
datasource = Path('data/dayabay-v0')
- storage |= load_parameters({'path': 'ibd' , 'load': datasource/'parameters/pdg2012.yaml'})
- storage |= load_parameters({'path': 'detector' , 'load': datasource/'parameters/detector_nprotons_correction.yaml'})
- storage |= load_parameters({'path': 'reactor' , 'load': datasource/'parameters/reactor_thermal_power_nominal.yaml'})
- storage |= load_parameters({'path': 'eres' , 'load': datasource/'parameters/detector_eres.yaml'})
+
+ with Graph() as g:
+ storage |= load_parameters({'path': 'ibd' , 'load': datasource/'parameters/pdg2012.yaml'})
+ storage |= load_parameters({'path': 'detector' , 'load': datasource/'parameters/detector_nprotons_correction.yaml'})
+ storage |= load_parameters({'path': 'reactor' , 'load': datasource/'parameters/reactor_thermal_power_nominal.yaml'})
+ storage |= load_parameters({'path': 'eres' , 'load': datasource/'parameters/detector_eres.yaml'})
# from pprint import pprint
# pprint(storage.object, sort_dicts=False)
@@ -63,3 +68,5 @@ def model_dayabay_v0():
print('Constants (latex)')
tex = storage['parameter.constant'].to_latex(columns=['path', 'value', 'label'])
print(tex)
+
+ savegraph(g, "output/dayabay_v0.dot")
--
GitLab