nnpdf-data 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

nnpdf_data/__init__.py

@@ -1,77 +1,22 @@
-from functools import lru_cache
 import pathlib
 
-import ruamel.yaml as yaml
-
 from ._version import __version__
+from .commondataparser import parse_new_metadata
+from .validphys_compatibility import legacy_to_new_map, legacy_to_new_mapping, new_to_legacy_map
 
 path_vpdata = pathlib.Path(__file__).parent
 path_commondata = path_vpdata / "commondata"
-
-# VP should not have access to this file, only to the products
-_path_legacy_mapping = path_commondata / "dataset_names.yml"
 theory_cards = path_vpdata / "theory_cards"
 
-_legacy_to_new_mapping_raw = yaml.YAML().load(_path_legacy_mapping)
-# Convert strings into a dictionary
-legacy_to_new_mapping = {
-    k: ({"dataset": v} if isinstance(v, str) else v) for k, v in _legacy_to_new_mapping_raw.items()
-}
-
-
-@lru_cache
-def legacy_to_new_map(dataset_name, sys=None):
-    """Find the new dataset name and variant corresponding to an old dataset
-    and systematics choice"""
-    if dataset_name not in legacy_to_new_mapping:
-        return dataset_name, None
-
-    new_name = legacy_to_new_mapping[dataset_name]
-    variant = new_name.get("variant")
-    new_name = new_name["dataset"]
-    if sys is not None:
-        if variant is None:
-            raise KeyError(
-                f"I cannot translate the combination of {dataset_name} and sys: {sys}. Please report this."
-            )
-        variant += f"_{sys}"
-
-    return new_name, variant
-
-
-@lru_cache
-def new_to_legacy_map(dataset_name, variant_used):
-    """Loop over the dictionary and find the right dataset.
-
-    Since it is posible to have more than 1 dataset mapped to the same new one,
-    returns a list of everything that matches.
-
-    This function will loop over the entire dictionary of mappings and selects
-    1. All datasets that match exactly what's in the runcard (dataset & variant): exact_matches
-    2. All datasets that match the dataset name: matches
-    If there are any `exact_matches`, it will return only those; otherwise, return all `matches`
-    if there are no `matches` at all, return None
-    """
-
-    matches = []
-    exact_matches = []
-
-    for old_name, new_info in legacy_to_new_mapping.items():
-        new_name = new_info["dataset"]
-        variant = new_info.get("variant")
 
-        if new_name == dataset_name:
-            matches.append(old_name)
-            # if it's a nuclear DIS data promote legacy to be legacy_dw
-            if "_DW_" in old_name and variant_used == "legacy":
-                variant = "legacy_dw"
+def load_dataset_metadata(dataset_name, variant=None):
+    """Given a dataset name, return the metadata"""
 
-            if variant_used == variant:
-                exact_matches.append(old_name)
+    # Compatibility with old nnpdf names, these two lines
+    # might disappear at any given point
+    if variant is None:
+        dataset_name, variant = legacy_to_new_map(dataset_name)
 
-    # If we found exact matches, return those and stop looking
-    if exact_matches:
-        return exact_matches
-    elif matches:
-        return matches
-    return None
+    setname, observable = dataset_name.rsplit("_", 1)
+    metadata_file = path_commondata / setname / "metadata.yaml"
+    return parse_new_metadata(metadata_file, observable, variant=variant)

nnpdf_data/_version.py

@@ -1 +1 @@
-__version__ = "4.0.9.post1772.dev0+604e6015c"
+__version__ = "0.0.2"

nnpdf_data/commondata/ATLAS_PH_13TEV/data_XSEC.yaml

@@ -0,0 +1,54 @@
+data_central:
+- 2.27318124e+03
+- 1.03060167e+03
+- 5.24219324e+02
+- 2.25822169e+02
+- 8.29410172e+01
+- 3.57822709e+01
+- 1.73424301e+01
+- 8.04542755e+00
+- 3.50689599e+00
+- 1.32750771e+00
+- 5.56675144e-01
+- 2.28021665e-01
+- 9.32813638e-02
+- 8.67157892e-03
+- 2.94648248e+03
+- 1.32864930e+03
+- 6.68062916e+02
+- 2.86003888e+02
+- 1.04956664e+02
+- 4.51198391e+01
+- 2.10088737e+01
+- 9.54611717e+00
+- 3.99013790e+00
+- 1.65668252e+00
+- 6.20319632e-01
+- 2.47423859e-01
+- 5.67952872e-02
+- 7.17810179e-03
+- 9.49375836e+02
+- 4.19455621e+02
+- 2.10686495e+02
+- 9.11711137e+01
+- 3.34379831e+01
+- 1.38974395e+01
+- 6.04588889e+00
+- 2.96851604e+00
+- 1.11503225e+00
+- 3.45174302e-01
+- 1.10401308e-01
+- 2.98410628e-02
+- 5.67142263e-03
+- 1.83047254e+03
+- 8.03284980e+02
+- 3.97683191e+02
+- 1.65270669e+02
+- 5.45840343e+01
+- 2.13516042e+01
+- 8.88513350e+00
+- 3.23701124e+00
+- 1.13758491e+00
+- 3.13408636e-01
+- 1.15459018e-01
+- 2.17765422e-02