Be more strict with requirements: scipy, pandas (#318)

* Be more strict with requirements: scipy, pandas

Closes #317

* Fix

* Fix

* moar fix

Author: dcherian

ci/minimal-requirements.yml

@@ -10,8 +10,8 @@ dependencies:
   - pytest-pretty
   - pytest-xdist
   - numpy==1.22
-  - scipy
+  - scipy==1.9.0
   - numpy_groupies==0.9.19
-  - pandas
+  - pandas==1.5
   - pooch
   - toolz

flox/core.py

@@ -23,7 +23,7 @@
 import numpy_groupies as npg
 import pandas as pd
 import toolz as tlz
-from scipy.sparse import csc_array
+from scipy.sparse import csc_array, csr_array
 
 from . import xrdtypes
 from .aggregate_flox import _prepare_for_flox
@@ -328,7 +328,7 @@ def invert(x) -> tuple[np.ndarray, ...]:
     #  - S is the existing set
     MIN_CONTAINMENT = 0.75  # arbitrary
     asfloat = bitmask.astype(float)
-    containment = ((asfloat.T @ asfloat) / chunks_per_label).tocsr()
+    containment = csr_array((asfloat.T @ asfloat) / chunks_per_label)
     mask = containment.data < MIN_CONTAINMENT
     containment.data[mask] = 0
     containment.eliminate_zeros()

pyproject.toml

@@ -39,10 +39,10 @@ test = ["netCDF4"]
 
 [build-system]
 requires = [
-    "pandas",
+    "pandas>=1.5",
     "numpy>=1.22",
     "numpy_groupies>=0.9.19",
-    "scipy",
+    "scipy>=1.9",
     "toolz",
     "setuptools>=61.0.0",
     "setuptools_scm[toml]>=7.0",

tests/test_core.py

@@ -876,8 +876,8 @@ def test_verify_complex_cohorts(chunksize: int) -> None:
     chunk_cohorts = find_group_cohorts(by - 1, (chunks,))
     chunks_ = np.sort(np.concatenate(tuple(chunk_cohorts.keys())))
     groups = np.sort(np.concatenate(tuple(chunk_cohorts.values())))
-    assert_equal(np.unique(chunks_), np.arange(len(chunks), dtype=int))
-    assert_equal(groups, np.arange(366, dtype=int))
+    assert_equal(np.unique(chunks_).astype(np.int64), np.arange(len(chunks), dtype=np.int64))
+    assert_equal(groups.astype(np.int64), np.arange(366, dtype=np.int64))
 
 
 @requires_dask