fix: writexml handles missing parameter configs for normfactor (#1819)

* In situations in writexml where the normfactor parameter config is missing inits or bounds, fall back to using default values instead. * Add tests that use workspaces with no parameter inits and no parameter bounds to cover this behavior.
scikit-hep · Mar 23, 2022 · 6d03b9e · 6d03b9e
1 parent 36e0d67
commit 6d03b9e
Show file tree

Hide file tree

Showing 4 changed files with 103 additions and 2 deletions.
diff --git a/src/pyhf/writexml.py b/src/pyhf/writexml.py
@@ -178,8 +178,8 @@ def build_modifier(spec, modifierspec, channelname, samplename, sampledata):
         high = 10
         for p in spec['measurements'][0]['config']['parameters']:
             if p['name'] == modifierspec['name']:
-                val = p['inits'][0]
-                low, high = p['bounds'][0]
+                val = p.get('inits', [val])[0]
+                low, high = p.get('bounds', [[low, high]])[0]
         attrs['Val'] = str(val)
         attrs['Low'] = str(low)
         attrs['High'] = str(high)

diff --git a/tests/test_export.py b/tests/test_export.py
@@ -428,3 +428,30 @@ def test_integer_data(datadir, mocker):
 
     channel = pyhf.writexml.build_channel(spec, channel_spec, {})
     assert channel
+
+
+@pytest.mark.parametrize(
+    "fname,val,low,high",
+    [
+        ('workspace_no_parameter_inits.json', '1', '-5', '5'),
+        ('workspace_no_parameter_bounds.json', '5', '0', '10'),
+    ],
+    ids=['no_inits', 'no_bounds'],
+)
+def test_issue1814(datadir, mocker, fname, val, low, high):
+    with open(datadir / fname) as spec_file:
+        spec = json.load(spec_file)
+
+    modifierspec = {'data': None, 'name': 'mu_sig', 'type': 'normfactor'}
+    channelname = None
+    samplename = None
+    sampledata = None
+
+    modifier = pyhf.writexml.build_modifier(
+        spec, modifierspec, channelname, samplename, sampledata
+    )
+    assert modifier is not None
+    assert sorted(modifier.keys()) == ['High', 'Low', 'Name', 'Val']
+    assert modifier.get('Val') == val
+    assert modifier.get('Low') == low
+    assert modifier.get('High') == high
diff --git a/tests/test_export/workspace_no_parameter_bounds.json b/tests/test_export/workspace_no_parameter_bounds.json
@@ -0,0 +1,37 @@
+{
+    "channels": [
+        {
+            "name": "ch",
+            "samples": [
+                {
+                    "data": [1000.0],
+                    "modifiers": [
+                        {"data": null, "name": "mu_sig", "type": "normfactor"},
+                        {
+                            "data": {"hi": 1.5, "lo": 0.5},
+                            "name": "unc",
+                            "type": "normsys"
+                        }
+                    ],
+                    "name": "signal"
+                }
+            ]
+        }
+    ],
+    "measurements": [
+        {
+            "config": {
+                "parameters": [
+                    {
+                        "name": "mu_sig",
+                        "inits": [5]
+                    }
+                ],
+                "poi": "mu_sig"
+            },
+            "name": "meas"
+        }
+    ],
+    "observations": [{"data": [1000], "name": "ch"}],
+    "version": "1.0.0"
+}
diff --git a/tests/test_export/workspace_no_parameter_inits.json b/tests/test_export/workspace_no_parameter_inits.json
@@ -0,0 +1,37 @@
+{
+    "channels": [
+        {
+            "name": "ch",
+            "samples": [
+                {
+                    "data": [1000.0],
+                    "modifiers": [
+                        {"data": null, "name": "mu_sig", "type": "normfactor"},
+                        {
+                            "data": {"hi": 1.5, "lo": 0.5},
+                            "name": "unc",
+                            "type": "normsys"
+                        }
+                    ],
+                    "name": "signal"
+                }
+            ]
+        }
+    ],
+    "measurements": [
+        {
+            "config": {
+                "parameters": [
+                    {
+                        "name": "mu_sig",
+                        "bounds": [[-5, 5]]
+                    }
+                ],
+                "poi": "mu_sig"
+            },
+            "name": "meas"
+        }
+    ],
+    "observations": [{"data": [1000], "name": "ch"}],
+    "version": "1.0.0"
+}