fix zip(..., strict=True) issue for Python 3.9

single-cell-data · Oct 4, 2024 · 679a282 · 679a282
1 parent 427f3e9
commit 679a282
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -69,7 +69,7 @@ plugins = "numpy.typing.mypy_plugin"
 
 [tool.ruff]
 lint.select = ["E", "F", "B", "I"]
-lint.ignore = ["E501"]  # line too long
+lint.ignore = ["E501", "B905"]  # line too long, zip without explicit `strict`
 lint.extend-select = ["I001"]  # unsorted-imports
 fix = true
 target-version = "py311"

diff --git a/tests/test_pytorch.py b/tests/test_pytorch.py
@@ -5,6 +5,7 @@
 
 from __future__ import annotations
 
+import sys
 from functools import partial
 from pathlib import Path
 from typing import Any, Callable, Optional, Sequence, Tuple, Union
@@ -936,7 +937,7 @@ def test_csr__merge(
             np.array_split(sp_coo.row, n_splits),
             np.array_split(sp_coo.col, n_splits),
             np.array_split(sp_coo.data, n_splits),
-            strict=False,
+            **(dict(strict=False) if sys.version_info > (3, 9) else {}),
         )
     ]
     _ncsr = _CSR_IO_Buffer.merge(