Add "--ignore-float-inf" for "correction summary" (#289)

* Add "--ignore-float-inf" for "correction summary"

* Pin dask for tests

Since dask-awkward does not have a release yet dask-contrib/dask-awkward#582

---------

Co-authored-by: Nicholas Smith <[email protected]>

Author: ftorrresd

pyproject.toml

@@ -46,6 +46,7 @@ test = [
   "awkward >=2.2.2;python_version>'3.7'",
   "awkward <2;python_version<='3.7'",
   "dask-awkward >=2024.1.1;python_version>'3.7'",
+  "dask <2025.4.0;python_version>'3.7'",
 ]
 dev = [
   "pytest >=4.6",

src/correctionlib/cli.py

@@ -1,6 +1,5 @@
-"""Command-line interface to correctionlib.
+"""Command-line interface to correctionlib."""
 
-"""
 import argparse
 import sys
 
@@ -77,6 +76,9 @@ def setup_validate(subparsers):
 
 
 def summary(console: Console, args: argparse.Namespace) -> int:
+    if args.ignore_float_inf:
+        schemav2.IGNORE_FLOAT_INF = True
+
     for file in args.files:
         console.rule(f"[blue]Corrections in file {file}")
         cset = model_auto(open_auto(file))
@@ -89,6 +91,11 @@ def setup_summary(subparsers):
         "summary", help="Print a summmary of the corrections"
     )
     parser.set_defaults(command=summary)
+    parser.add_argument(
+        "--ignore-float-inf",
+        action="store_true",
+        help="Disable errors for use of float infinities in bin edges (as required in v2.6 and later)",
+    )
     parser.add_argument("files", nargs="+", metavar="FILE")
     return parser