diff --git a/pyproject.toml b/pyproject.toml index c922b9bd..87111848 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -110,7 +110,7 @@ include-package-data = false package-dir = {"" = "src"} [tool.setuptools.package-data] -"polarimetry.data" = [ +"polarimetry.lhcb" = [ "model-definitions.yaml", "particle-definitions.yaml", ] diff --git a/src/polarimetry/data/model-definitions.yaml b/src/polarimetry/lhcb/model-definitions.yaml similarity index 100% rename from src/polarimetry/data/model-definitions.yaml rename to src/polarimetry/lhcb/model-definitions.yaml diff --git a/src/polarimetry/data/particle-definitions.yaml b/src/polarimetry/lhcb/particle-definitions.yaml similarity index 100% rename from src/polarimetry/data/particle-definitions.yaml rename to src/polarimetry/lhcb/particle-definitions.yaml diff --git a/src/polarimetry/lhcb/particle.py b/src/polarimetry/lhcb/particle.py index cee35d33..72c23a19 100644 --- a/src/polarimetry/lhcb/particle.py +++ b/src/polarimetry/lhcb/particle.py @@ -74,7 +74,7 @@ class ResonanceJSON(TypedDict): __PARTICLE_DATABASE = load_particles( - Path(__file__).parent.parent / "data/particle-definitions.yaml" + Path(__file__).parent.parent / "lhcb/particle-definitions.yaml" ) Λc = __PARTICLE_DATABASE["Lambda_c+"]