mergeron 2025.739265.0__py3-none-any.whl → 2025.739290.0__py3-none-any.whl

mergeron/gen/data_generation_functions.py

@@ -4,22 +4,28 @@ Non-public functions called in data_generation.py
 
 from __future__ import annotations
 
-from collections.abc import Sequence
 from typing import Literal
 
 import numpy as np
 from attrs import evolve
 from numpy.random import SeedSequence
 
-from .. import DEFAULT_REC_RATIO, VERSION, ArrayDouble, RECForm  # noqa: TID252
+from .. import (  # noqa: TID252
+    DEFAULT_REC_RATIO,
+    NTHREADS,
+    VERSION,
+    ArrayDouble,
+    ArrayFloat,
+    RECForm,
+)
 from ..core.empirical_margin_distribution import mgn_data_resampler  # noqa: TID252
 from ..core.pseudorandom_numbers import (  # noqa: TID252
+    DEFAULT_BETA_DIST_PARMS,
     DEFAULT_DIST_PARMS,
     MultithreadedRNG,
     prng,
 )
 from . import (
-    DEFAULT_EMPTY_ARRAY,
     DEFAULT_FCOUNT_WTS,
     FM2Constraint,
     MarginDataSample,
@@ -27,7 +33,6 @@ from . import (
     PCMSpec,
     PriceDataSample,
     PriceSpec,
-    SeedSequenceData,
     ShareDataSample,
     ShareSpec,
     SHRDistribution,
@@ -41,8 +46,8 @@ def gen_share_data(
     _sample_size: int,
     _share_spec: ShareSpec,
     _fcount_rng_seed_seq: SeedSequence | None,
-    _mktshr_rng_seed_seq: SeedSequence,
-    _nthreads: int = 16,
+    _mktshr_rng_seed_seq: SeedSequence | None = None,
+    _nthreads: int = NTHREADS,
     /,
 ) -> ShareDataSample:
     """Helper function for generating share data.
@@ -70,6 +75,12 @@ def gen_share_data(
     )
 
     _ssz = _sample_size
+    if not len(_dist_parms_mktshr):
+        _dist_parms_mktshr = (
+            DEFAULT_DIST_PARMS
+            if _dist_type_mktshr == "Uniform"
+            else np.ones(1 + len(_firm_count_prob_wts), float)
+        )
 
     if _dist_type_mktshr == SHRDistribution.UNI:
         _mkt_share_sample = gen_market_shares_uniform(
@@ -80,9 +91,9 @@ def gen_share_data(
         _firm_count_prob_wts = (
             None
             if _firm_count_prob_wts is None
-            else np.array(_firm_count_prob_wts, dtype=np.float64)
+            else np.array(_firm_count_prob_wts, float)
         )
-        _mkt_share_sample = gen_market_shares_dirichlet_multimarket(
+        _mkt_share_sample = _gen_market_shares_dirichlet_multimarket(
             _ssz,
             _recapture_form,
             _dist_type_mktshr,
@@ -113,10 +124,10 @@ def gen_share_data(
 
 
 def gen_market_shares_uniform(
-    _s_size: int = 10**6,
-    _dist_parms_mktshr: ArrayDouble | None = DEFAULT_DIST_PARMS,
+    _s_size: int,
+    _dist_parms_mktshr: ArrayDouble,
     _mktshr_rng_seed_seq: SeedSequence | None = None,
-    _nthreads: int = 16,
+    _nthreads: int = NTHREADS,
     /,
 ) -> ShareDataSample:
     """Generate merging-firm shares from Uniform distribution on the 3-D simplex.
@@ -138,9 +149,8 @@ def gen_market_shares_uniform(
 
     """
 
-    _frmshr_array = np.empty((_s_size, 2), dtype=np.float64)
+    _frmshr_array: ArrayDouble = np.empty((_s_size, 2), float)
 
-    _dist_parms_mktshr = _dist_parms_mktshr or DEFAULT_DIST_PARMS
     _mrng = MultithreadedRNG(
         _frmshr_array,
         dist_type="Uniform",
@@ -151,13 +161,14 @@ def gen_market_shares_uniform(
     _mrng.fill()
     # Convert draws on U[0, 1] to Uniformly-distributed draws on simplex, s_1 + s_2 <= 1
     _frmshr_array.sort(axis=1)
-    _frmshr_array = np.column_stack((
-        _frmshr_array[:, 0],
-        _frmshr_array[:, 1] - _frmshr_array[:, 0],
-    ))
+
+    _frmshr_array = np.array(
+        (_frmshr_array[:, 0], _frmshr_array[:, 1] - _frmshr_array[:, 0]),
+        _frmshr_array.dtype,
+    ).T  # faster than np.stack() and variants
 
     # Keep only share combinations representing feasible mergers
-    # This is a no-op for 64-bit floats, but is necessary for 32-bit floats
+    # This is a no-op for 64-bit floats, but is necessary for smaller floats
     _frmshr_array = _frmshr_array[_frmshr_array.min(axis=1) > 0]
 
     # Let a third column have values of "np.nan", so HHI calculations return "np.nan"
@@ -172,7 +183,7 @@ def gen_market_shares_uniform(
     )
 
     # This array is meant to be ignored, so a sentinel value is fine
-    _fcounts.fill(-9999)
+    _fcounts.fill(-1)
 
     _nth_firm_share.fill(np.nan)
     _aggregate_purchase_prob.fill(np.nan)
@@ -182,15 +193,15 @@ def gen_market_shares_uniform(
     )
 
 
-def gen_market_shares_dirichlet_multimarket(
-    _s_size: int = 10**6,
-    _recapture_form: RECForm = RECForm.INOUT,
-    _dist_type_dir: SHRDistribution = SHRDistribution.DIR_FLAT,
-    _dist_parms_dir: ArrayDouble | None = None,
-    _firm_count_wts: ArrayDouble | None = None,
+def _gen_market_shares_dirichlet_multimarket(
+    _s_size: int,
+    _recapture_form: RECForm,
+    _dist_type_dir: SHRDistribution,
+    _dist_parms_dir: ArrayDouble,
+    _firm_count_wts: ArrayDouble,
     _fcount_rng_seed_seq: SeedSequence | None = None,
     _mktshr_rng_seed_seq: SeedSequence | None = None,
-    _nthreads: int = 16,
+    _nthreads: int = NTHREADS,
     /,
 ) -> ShareDataSample:
     """Dirichlet-distributed shares with multiple firm-counts.
@@ -229,10 +240,11 @@ def gen_market_shares_dirichlet_multimarket(
 
     """
 
-    # _firm_count_wts: ArrayDouble = (
-    #     DEFAULT_FCOUNT_WTS if _firm_count_wts is None else _firm_count_wts
-    # )
-    _firm_count_wts = DEFAULT_FCOUNT_WTS if _firm_count_wts is None else _firm_count_wts
+    _firm_count_wts = (
+        DEFAULT_FCOUNT_WTS
+        if _firm_count_wts is None or not len(_firm_count_wts)
+        else _firm_count_wts
+    )
 
     _min_choice_wt = 0.03 if _dist_type_dir == SHRDistribution.DIR_FLAT_CONSTR else 0.00
     _fcount_keys, _choice_wts = zip(
@@ -250,7 +262,7 @@ def gen_market_shares_dirichlet_multimarket(
 
     _fc_max = _fcount_keys[-1]
     _dir_alphas_full = (
-        [1.0] * _fc_max if _dist_parms_dir is None else _dist_parms_dir[:_fc_max]
+        _dist_parms_dir[:_fc_max] if len(_dist_parms_dir) else [1.0] * _fc_max
     )
     if _dist_type_dir == SHRDistribution.DIR_ASYM:
         _dir_alphas_full = [2.0] * 6 + [1.5] * 5 + [1.25] * min(7, _fc_max)
@@ -261,12 +273,12 @@ def gen_market_shares_dirichlet_multimarket(
             _dat = [2.5] * 2
             if _fcv > len(_dat):
                 _dat += [1.0 / (_fcv - 2)] * (_fcv - 2)
-            return np.array(_dat, dtype=np.float64)
+            return np.array(_dat, float)
 
     else:
 
         def _gen_dir_alphas(_fcv: int) -> ArrayDouble:
-            return np.array(_dir_alphas_full[:_fcv], dtype=np.float64)
+            return np.array(_dir_alphas_full[:_fcv], float)
 
     _fcounts = prng(_fcount_rng_seed_seq).choice(
         _fcount_keys, size=(_s_size, 1), p=_choice_wts
@@ -281,7 +293,7 @@ def gen_market_shares_dirichlet_multimarket(
     _aggregate_purchase_prob, _nth_firm_share = (
         np.empty((_s_size, 1)) for _ in range(2)
     )
-    _mktshr_array = np.empty((_s_size, _fc_max), dtype=np.float64)
+    _mktshr_array = np.empty((_s_size, _fc_max), float)
     for _f_val, _f_sseq in zip(_fcount_keys, _mktshr_seed_seq_ch, strict=True):
         _fcounts_match_rows = np.where(_fcounts == _f_val)[0]
         _dir_alphas_test = _gen_dir_alphas(_f_val)
@@ -327,10 +339,10 @@ def gen_market_shares_dirichlet_multimarket(
 
 def gen_market_shares_dirichlet(
     _dir_alphas: ArrayDouble,
-    _s_size: int = 10**6,
-    _recapture_form: RECForm = RECForm.INOUT,
+    _s_size: int,
+    _recapture_form: RECForm,
     _mktshr_rng_seed_seq: SeedSequence | None = None,
-    _nthreads: int = 16,
+    _nthreads: int = NTHREADS,
     /,
 ) -> ShareDataSample:
     """Dirichlet-distributed shares with fixed firm-count.
@@ -372,7 +384,7 @@ def gen_market_shares_dirichlet(
         else SeedSequence(pool_size=8)
     )
 
-    _mktshr_array = np.empty((_s_size, len(_dir_alphas)), dtype=np.float64)
+    _mktshr_array = np.empty((_s_size, len(_dir_alphas)), float)
     _mrng = MultithreadedRNG(
         _mktshr_array,
         dist_type="Dirichlet",
@@ -396,11 +408,11 @@ def gen_market_shares_dirichlet(
         )
 
     # If recapture_form == 'inside_out', further calculations downstream
-    _aggregate_purchase_prob = np.empty((_s_size, 1), dtype=np.float64)
+    _aggregate_purchase_prob = np.empty((_s_size, 1), float)
     _aggregate_purchase_prob.fill(np.nan)
     if _recapture_form == RECForm.OUTIN:
         _aggregate_purchase_prob = 1 - _mktshr_array[:, [-1]]  # type: ignore
-        _mktshr_array = _mktshr_array[:, :-1] / _aggregate_purchase_prob  # type: ignore
+        _mktshr_array = _mktshr_array[:, :-1] / _aggregate_purchase_prob
 
     return ShareDataSample(
         _mktshr_array,
@@ -414,7 +426,7 @@ def gen_divr_array(
     _recapture_form: RECForm,
     _recapture_ratio: float | None,
     _frmshr_array: ArrayDouble,
-    _aggregate_purchase_prob: ArrayDouble = DEFAULT_EMPTY_ARRAY,
+    _aggregate_purchase_prob: ArrayDouble,
     /,
 ) -> ArrayDouble:
     """
@@ -484,8 +496,8 @@ def gen_margin_price_data(
     _price_spec: PriceSpec,
     _hsr_filing_test_type: SSZConstant,
     _pcm_rng_seed_seq: SeedSequence,
-    _pr_rng_seed_seq: SeedSequence | None = None,
-    _nthreads: int = 16,
+    _pr_rng_seed_seq: SeedSequence | None,
+    _nthreads: int = NTHREADS,
     /,
 ) -> tuple[MarginDataSample, PriceDataSample]:
     """Generate margin and price data for mergers in the sample.
@@ -532,10 +544,8 @@ def gen_margin_price_data(
         np.empty_like(_frmshr_array), np.ones(len(_frmshr_array)) == 0
     )
 
-    _price_array, _price_ratio_array = (
-        np.ones_like(_frmshr_array, np.float64),
-        np.empty_like(_frmshr_array, np.float64),
-    )
+    _nth_firm_price: ArrayDouble
+    _price_array: ArrayDouble = np.ones_like(_frmshr_array, np.float64)
 
     _pr_max_ratio = 5.0
     match _price_spec:
@@ -555,7 +565,7 @@ def gen_margin_price_data(
                 1 + np.arange(_pr_max_ratio), size=(len(_frmshr_array), 3)
             )
             _price_array = _price_array_gen[:, :2]
-            _nth_firm_price = _price_array_gen[:, [2]]  # type: ignore
+            _nth_firm_price = _price_array_gen[:, [2]]
             # del _price_array_gen
         case PriceSpec.CSY:
             # TODO:
@@ -616,7 +626,7 @@ def gen_margin_price_data(
             _nthreads,
         )
 
-    _price_array = _price_array.astype(np.float64)
+    # _price_array = _price_array.astype(np.float64)
     _rev_array = _price_array * _frmshr_array
     _nth_firm_rev = _nth_firm_price * _nth_firm_share
 
@@ -670,8 +680,8 @@ def _gen_margin_data(
     _price_array: ArrayDouble,
     _aggregate_purchase_prob: ArrayDouble,
     _pcm_spec: PCMSpec,
-    _pcm_rng_seed_seq: SeedSequence,
-    _nthreads: int = 16,
+    _pcm_rng_seed_seq: SeedSequence | None,
+    _nthreads: int = NTHREADS,
     /,
 ) -> MarginDataSample:
     _dist_type_pcm, _dist_firm2_pcm, _dist_parms_pcm = (
@@ -680,12 +690,13 @@ def _gen_margin_data(
     )
 
     _pcm_array = (
-        np.empty((len(_frmshr_array), 1), dtype=np.float64)
+        np.empty((len(_frmshr_array), 1), float)
         if _pcm_spec.firm2_pcm_constraint == FM2Constraint.SYM
-        else np.empty_like(_frmshr_array, dtype=np.float64)
+        else np.empty_like(_frmshr_array, float)
     )
 
-    _beta_min, _beta_max = [None] * 2  # placeholder
+    _dist_parms: ArrayFloat
+    _beta_min, _beta_max = [0.0] * 2  # placeholder
     if _dist_type_pcm == PCMDistribution.EMPR:
         _pcm_array = mgn_data_resampler(
             _pcm_array.shape, seed_sequence=_pcm_rng_seed_seq
@@ -696,11 +707,11 @@ def _gen_margin_data(
             _dist_type = "Beta"
             _dist_parms_pcm = (
                 (
-                    np.array([0, 1, 0, 1], np.float64)
-                    if _dist_parms_pcm == PCMDistribution.BETA_BND
-                    else np.ones(2, np.float64)
+                    np.array([0.5, 1, 0, 1], float)
+                    if _dist_type_pcm == PCMDistribution.BETA_BND
+                    else DEFAULT_BETA_DIST_PARMS
                 )
-                if _dist_parms_pcm is None
+                if _dist_parms_pcm is None or not len(_dist_parms_pcm)
                 else _dist_parms_pcm
             )
             _dist_parms = beta_located_bound(_dist_parms_pcm)
@@ -708,7 +719,9 @@ def _gen_margin_data(
         else:
             _dist_type = "Uniform"
             _dist_parms = (
-                DEFAULT_DIST_PARMS if _dist_parms_pcm is None else _dist_parms_pcm
+                DEFAULT_DIST_PARMS
+                if _dist_parms_pcm is None or not len(_dist_parms_pcm)
+                else _dist_parms_pcm
             )
 
         _pcm_rng = MultithreadedRNG(
@@ -779,7 +792,7 @@ def _beta_located(
     """
 
     _mul = -1 + _mu * (1 - _mu) / _sigma**2
-    return np.array([_mu * _mul, (1 - _mu) * _mul], dtype=np.float64)
+    return np.array([_mu * _mul, (1 - _mu) * _mul], float)
 
 
 def beta_located_bound(_dist_parms: ArrayDouble, /) -> ArrayDouble:
@@ -812,65 +825,3 @@ def beta_located_bound(_dist_parms: ArrayDouble, /) -> ArrayDouble:
 
     _bmu, _bsigma, _bmin, _bmax = _dist_parms
     return _beta_located((_bmu - _bmin) / (_bmax - _bmin), _bsigma / (_bmax - _bmin))
-
-
-def parse_seed_seq_list(
-    _sseq_list: Sequence[SeedSequence] | None,
-    _mktshr_dist_type: SHRDistribution,
-    _price_spec: PriceSpec,
-    /,
-) -> SeedSequenceData:
-    """Initialize RNG seed sequences to ensure independence of distinct random streams.
-
-    The tuple of SeedSequences, is parsed in the following order
-    for generating the relevant random variates:
-    1.) quantity shares
-    2.) price-cost margins
-    3.) firm-counts, if :code:`MarketSpec.share_spec.dist_type` is a Dirichlet distribution
-    4.) prices, if :code:`MarketSpec.price_spec ==`:attr:`mergeron.gen.PriceSpec.ZERO`.
-
-
-
-    Parameters
-    ----------
-    _sseq_list
-        List of RNG seed sequences
-
-    _mktshr_dist_type
-        Market share distribution type
-
-    _price_spec
-        Price specification
-
-    Returns
-    -------
-        Seed sequence data
-
-    """
-    _seed_count = 2 if _mktshr_dist_type == SHRDistribution.UNI else 3
-    _seed_count += 1 if _price_spec == PriceSpec.ZERO else 0
-
-    _fcount_rng_seed_seq: SeedSequence | None = None
-    _pr_rng_seed_seq: SeedSequence | None = None
-
-    _sseq_list = (
-        _sseq_list
-        if _sseq_list
-        else tuple(SeedSequence(pool_size=8) for _ in range(_seed_count))
-    )
-
-    if (_l := len(_sseq_list)) < _seed_count:
-        raise ValueError(
-            f"Seed sequence list must contain {_seed_count} seed sequences; "
-            f"only {_l} given."
-        )
-
-    _mktshr_rng_seed_seq, _pcm_rng_seed_seq = _sseq_list[:2]
-    _fcount_rng_seed_seq = (
-        None if _mktshr_dist_type == SHRDistribution.UNI else _sseq_list[2]
-    )
-    _pr_rng_seed_seq = _sseq_list[-1] if _price_spec == PriceSpec.ZERO else None
-
-    return SeedSequenceData(
-        _mktshr_rng_seed_seq, _pcm_rng_seed_seq, _fcount_rng_seed_seq, _pr_rng_seed_seq
-    )

mergeron/gen/enforcement_stats.py

@@ -9,7 +9,7 @@ from collections.abc import Mapping
 import numpy as np
 from scipy.interpolate import interp1d  # type: ignore
 
-from .. import VERSION, ArrayBIGINT  # noqa: TID252
+from .. import VERSION, ArrayBIGINT, this_yaml  # noqa: TID252
 from ..core import ftc_merger_investigations_data as fid  # noqa: TID252
 from . import INVResolution
 
@@ -291,10 +291,10 @@ def enf_cnts_byconczone(_cnts_array: ArrayBIGINT, /) -> ArrayBIGINT:
     # in both cases does make life easier
     _ndim_in = 2
     _nkeys = 3
-    _cnts_byhhipostanddelta = -1 * np.ones(
-        _nkeys + _cnts_array.shape[1] - _ndim_in, dtype=np.int64
+    _cnts_byhhipostanddelta: ArrayBIGINT = -1 * np.ones(
+        _nkeys + _cnts_array.shape[1] - _ndim_in, dtype=int
     )
-    _cnts_byconczone = -1 * np.ones_like(_cnts_byhhipostanddelta)
+    _cnts_byconczone: ArrayBIGINT = -1 * np.ones_like(_cnts_byhhipostanddelta)
     for _hhi_zone_post_lim in HHI_POST_ZONE_KNOTS[:-1]:
         _level_test = _hhi_zone_post_ranged == _hhi_zone_post_lim
 
@@ -318,7 +318,7 @@ def enf_cnts_byconczone(_cnts_array: ArrayBIGINT, /) -> ArrayBIGINT:
                         *_zone_val,
                         *np.einsum("ij->j", _cnts_array[:, _ndim_in:][_conc_test]),
                     ),
-                    dtype=np.int64,
+                    dtype=int,
                 ),
             ))
     _cnts_byhhipostanddelta = _cnts_byhhipostanddelta[1:]
@@ -342,13 +342,37 @@ def enf_cnts_byconczone(_cnts_array: ArrayBIGINT, /) -> ArrayBIGINT:
                         "ij->j", _cnts_byhhipostanddelta[_hhi_zone_test][:, _nkeys:]
                     ),
                 ),
-                dtype=np.int64,
+                dtype=int,
             ),
         ))
 
     return _cnts_byconczone[1:]
 
 
+for _typ in (
+    IndustryGroup,
+    OtherEvidence,
+    StatsGrpSelector,
+    StatsReturnSelector,
+    SortSelector,
+):
+    #  NOTE: If additional enums are defined in this module,
+    #  add themn to the list above
+
+    _, _ = (
+        this_yaml.representer.add_representer(
+            _typ,
+            lambda _r, _d: _r.represent_scalar(f"!{_d.__class__.__name__}", _d.name),
+        ),
+        this_yaml.constructor.add_constructor(
+            f"!{_typ.__name__}",
+            lambda _c, _n, /: getattr(
+                globals().get(_n.tag.lstrip("!")), _c.construct_scalar(_n)
+            ),
+        ),
+    )
+
+
 if __name__ == "__main__":
     print(
         "This module provides methods to aggregate statistics on merger enforcement patterns for reporting."

mergeron/gen/upp_tests.py

@@ -26,10 +26,9 @@ from .. import (  # noqa
 )
 from ..core import guidelines_boundaries as gbl  # noqa: TID252
 from . import (
-    DEFAULT_EMPTY_ARRAY,
     DataclassInstance,
     INVResolution,
-    MarketDataSample,
+    MarketSampleData,
     UPPTestRegime,
     UPPTestsCounts,
     UPPTestsRaw,
@@ -52,7 +51,7 @@ class INVRESCntsArgs(TypedDict, total=False):
 
 
 def compute_upp_test_counts(
-    _market_data_sample: MarketDataSample,
+    _market_data_sample: MarketSampleData,
     _upp_test_parms: gbl.HMGThresholds,
     _upp_test_regime: UPPTestRegime,
     /,
@@ -96,7 +95,7 @@ def compute_upp_test_counts(
     if _firmcounts_list is not None and np.all(_firmcounts_list >= 0):
         _max_firmcount = max(_firmcounts_list)
 
-        _enf_cnts_sim_byfirmcount_array = -1 * np.ones(_stats_rowlen, np.int64)
+        _enf_cnts_sim_byfirmcount_array: ArrayBIGINT = -1 * np.ones(_stats_rowlen, int)
         for _firmcount in np.arange(2, _max_firmcount + 1):
             _firmcount_test = _fcounts == _firmcount
 
@@ -123,7 +122,7 @@ def compute_upp_test_counts(
 
     # Clearance/enforcement counts --- by delta
     _hhi_delta_ranged = esl.hhi_delta_ranger(_hhi_delta)
-    _enf_cnts_sim_bydelta_array = -1 * np.ones(_stats_rowlen, np.int64)
+    _enf_cnts_sim_bydelta_array: ArrayBIGINT = -1 * np.ones(_stats_rowlen, int)
     for _hhi_delta_lim in esl.HHI_DELTA_KNOTS[:-1]:
         _hhi_delta_test = _hhi_delta_ranged == _hhi_delta_lim
 
@@ -190,7 +189,7 @@ def compute_upp_test_counts(
 
 
 def compute_upp_test_arrays(
-    _market_data: MarketDataSample,
+    _market_data: MarketSampleData,
     _upp_test_parms: gbl.HMGThresholds,
     _sim_test_regime: UPPTestRegime,
     /,
@@ -279,7 +278,7 @@ def _compute_test_array_seq(
             UPPAggrSelector.OSA,
             UPPAggrSelector.OSD,
         )
-        else DEFAULT_EMPTY_ARRAY
+        else np.array([0.5, 0.5], float)
     )
 
     match _aggregator:
@@ -331,11 +330,11 @@ def initialize_hd5(
     _h5_title = f"HMG version: {_hmg_pub_year}; Test regime: {_test_regime}"
     if _h5_path.is_file():
         _h5_path.unlink()
-    _h5_file = ptb.open_file(_h5_path, mode="w", title=_h5_title)  # pyright: ignore
+    _h5_file = ptb.open_file(_h5_path, mode="w", title=_h5_title)
     _save_data_to_file: SaveData = (True, _h5_file, _h5_file.root)
     _next_subgroup_name_root = "enf_{}_{}_{}_{}".format(
         _hmg_pub_year,
-        *(getattr(_test_regime, _f.name).name for _f in _test_regime.__attrs_attrs__),  # pyright: ignore
+        *(getattr(_test_regime, _f.name).name for _f in _test_regime.__attrs_attrs__),
     )
     return _save_data_to_file, _next_subgroup_name_root
 
@@ -383,7 +382,7 @@ def save_array_to_hdf5(
         _h5_array_name,
         atom=ptb.Atom.from_dtype(_array_obj.dtype),
         shape=_array_obj.shape,
-        filters=ptb.Filters(complevel=3, complib="blosc:lz4hc", fletcher32=True),  # pyright: ignore
+        filters=ptb.Filters(complevel=3, complib="blosc:lz4hc", fletcher32=True),
     )
     _h5_array[:] = _array_obj
 

{mergeron-2025.739265.0.dist-info → mergeron-2025.739290.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mergeron
-Version: 2025.739265.0
+Version: 2025.739290.0
 Summary: Analyze merger enforcement policy using Python
 License: MIT
 Keywords: merger policy analysis,merger guidelines,merger screening,policy presumptions,concentration standards,upward pricing pressure,GUPPI
@@ -30,6 +30,7 @@ Requires-Dist: matplotlib (>=3.8)
 Requires-Dist: mpmath (>=1.3)
 Requires-Dist: msgpack (>=1.0)
 Requires-Dist: msgpack-numpy (>=0.4)
+Requires-Dist: ruamel-yaml (>=0.18.10,<0.19.0)
 Requires-Dist: scipy (>=1.12)
 Requires-Dist: sympy (>=1.12)
 Requires-Dist: tables (>=3.10.1)

mergeron-2025.739290.0.dist-info/RECORD

@@ -0,0 +1,23 @@
+mergeron/__init__.py,sha256=CJgJmTbFDmz9oNPWROam9rB1pikAKF1SdqsL9putg5M,3035
+mergeron/core/__init__.py,sha256=qYRA1D4Gx-WSkICqiueWLWPedkWmb_CIn7fgvN7unk4,182
+mergeron/core/empirical_margin_distribution.py,sha256=byO11ROGVQcmbZVWdVJkJ906WnHhx7jcoL5BbxPmX7I,8688
+mergeron/core/ftc_merger_investigations_data.py,sha256=9AT-phfkn0NHJvn9lcpbH8ExWZStYW47KQjNjDGS3qc,28709
+mergeron/core/guidelines_boundaries.py,sha256=U2SaG5SMV51nmBNe8n94RQPJCHyxFGKaQ5ncA-COe5g,15360
+mergeron/core/guidelines_boundary_functions.py,sha256=4vMz5DQ72fgj7leUlVJvXBrxdZAeDJP7FoGBEVXDdeQ,34866
+mergeron/core/guidelines_boundary_functions_extra.py,sha256=qL1zT-Xgj2qD2EyBRR87APoVc47MkGM02xYQhNgwEvg,11216
+mergeron/core/pseudorandom_numbers.py,sha256=qmXErMPUlP769cItwxHBw1qW2kYMjcQ26P4Dr7McOnQ,10372
+mergeron/data/__init__.py,sha256=KtjBlZOl7jwBCAUhrTJB9PdrN39YLYytNiSUSM_gRmA,62
+mergeron/data/damodaran_margin_data.xls,sha256=Qggl1p5nkOMJI8YUXhkwXQRz-OhRSqBTzz57N0JQyYA,79360
+mergeron/data/damodaran_margin_data_dict.msgpack,sha256=sr6s4L69kposEpzGI7jpPb4ULz0UpY-bEYfeNi6UlRA,57621
+mergeron/data/ftc_invdata.msgpack,sha256=WBFHgi7Ld4R-h2zL2Zc3TOIlKqVrbVFMH1LoI4-T-M0,264664
+mergeron/demo/__init__.py,sha256=KtjBlZOl7jwBCAUhrTJB9PdrN39YLYytNiSUSM_gRmA,62
+mergeron/demo/visualize_empirical_margin_distribution.py,sha256=kDwPfhsBjsQdYPyhe4KZG2guB_xHhPaCrvn8fezIz4M,2354
+mergeron/gen/__init__.py,sha256=t1RswXk-L6olOhj5XqPekd8je8Za7NUFuSo3fuM2Wts,21787
+mergeron/gen/data_generation.py,sha256=0Z-TvQANvEjrmXm9QKdmcm9mieWx2iF91K_7Ij5B0zI,16254
+mergeron/gen/data_generation_functions.py,sha256=LwqbCYSqK2PfFrHhZaTOaF7UtlsVnhm0Ykmfgf3P-G4,27187
+mergeron/gen/enforcement_stats.py,sha256=z_Kjw1OwLue5B-tKFlmEoyGDusi9D6c77foRCmiNGqQ,11532
+mergeron/gen/upp_tests.py,sha256=VTh0rDd0cJ70MwkvRDYx27LWMaA1BPThXfd3D1HKX-E,12489
+mergeron/py.typed,sha256=frcCV1k9oG9oKj3dpUqdJg1PxRT2RSN_XKdLCPjaYaY,2
+mergeron-2025.739290.0.dist-info/METADATA,sha256=-Ib5QC5-mq2ItR_bS1s08BRGknEVbsPPfiDIJavzQbU,14541
+mergeron-2025.739290.0.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+mergeron-2025.739290.0.dist-info/RECORD,,

mergeron-2025.739265.0.dist-info/RECORD

@@ -1,23 +0,0 @@
-mergeron/__init__.py,sha256=TA29lm9xq1AOShjbOs4dXujCClb_MMCLlKN6ej8VC08,1581
-mergeron/core/__init__.py,sha256=jPGd0okmvNOdWnTu4biR4hqnL29IERaY4Olv8mS02ko,188
-mergeron/core/empirical_margin_distribution.py,sha256=uDgXXNNL4ln_f1rnvfOxrpokeuha6qQXpi0c7ZGE58U,8726
-mergeron/core/ftc_merger_investigations_data.py,sha256=_Qno1_oWcHmDd60j8YX6BLUSLeAn-DU6KpvMOcbCIaE,28728
-mergeron/core/guidelines_boundaries.py,sha256=ci6bNVNKGynCnXFoQXit0jyhpQW8hw-Y37DpozuQETg,15245
-mergeron/core/guidelines_boundary_functions.py,sha256=SoTamDsKrThqDfoNTHeGp2ntiwhzscWmkAbIWrBtUq4,34885
-mergeron/core/guidelines_boundary_functions_extra.py,sha256=ytTWxYk7FkGLQubkQXrniwP_2ZlYssp4rmOxy4_7480,11292
-mergeron/core/pseudorandom_numbers.py,sha256=VepI_YU5oMkTExjyN9OyfiB3UmTDR2cq4N1DAmYZbm4,9233
-mergeron/data/__init__.py,sha256=KtjBlZOl7jwBCAUhrTJB9PdrN39YLYytNiSUSM_gRmA,62
-mergeron/data/damodaran_margin_data.xls,sha256=Qggl1p5nkOMJI8YUXhkwXQRz-OhRSqBTzz57N0JQyYA,79360
-mergeron/data/damodaran_margin_data_dict.msgpack,sha256=sr6s4L69kposEpzGI7jpPb4ULz0UpY-bEYfeNi6UlRA,57621
-mergeron/data/ftc_invdata.msgpack,sha256=WBFHgi7Ld4R-h2zL2Zc3TOIlKqVrbVFMH1LoI4-T-M0,264664
-mergeron/demo/__init__.py,sha256=KtjBlZOl7jwBCAUhrTJB9PdrN39YLYytNiSUSM_gRmA,62
-mergeron/demo/visualize_empirical_margin_distribution.py,sha256=_1SWKqFJNqKV_yA3v2tmzl8Th3CC3SXpb6VGiwGGuN0,2373
-mergeron/gen/__init__.py,sha256=6A1HVewEEwLo4JtZZ61ShzhU2phEeNfTxmTPlpi5LXY,17215
-mergeron/gen/data_generation.py,sha256=nzCWVJBXYxloR6J6MamFM0MUbDzguhARCAEksYkX1Dw,16872
-mergeron/gen/data_generation_functions.py,sha256=cIYkE4DcBXdEbEIeCk2lG9rLKblhdYZPDMJdGMS2dKE,29022
-mergeron/gen/enforcement_stats.py,sha256=ZjrV_VkFMF0D1myc-fj-W99M1EhJMA9-nCfyE5g9e54,10890
-mergeron/gen/upp_tests.py,sha256=bQfxigqHc8bdsd5SjMyKKa07weTEcAdJvLCiTrhgzAE,12547
-mergeron/py.typed,sha256=frcCV1k9oG9oKj3dpUqdJg1PxRT2RSN_XKdLCPjaYaY,2
-mergeron-2025.739265.0.dist-info/METADATA,sha256=nkTdrTqjVT4tc6ljYRuCFykz0X-TsABN3zxyEe9ALnE,14494
-mergeron-2025.739265.0.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-mergeron-2025.739265.0.dist-info/RECORD,,