| |
| """Patch chumpy 0.70 for Python 3.11 and NumPy 2.x. |
| |
| Run this after: |
| |
| pip install --no-build-isolation chumpy==0.70 |
| |
| The script intentionally avoids importing chumpy, because the unpatched package |
| can fail during import on modern Python/NumPy versions. |
| """ |
|
|
| from __future__ import annotations |
|
|
| import importlib.util |
| from pathlib import Path |
|
|
|
|
| GETARGSPEC_OLD = "inspect.getargspec(" |
| GETARGSPEC_NEW = "inspect.getfullargspec(" |
|
|
| NUMPY_IMPORT_OLD = ( |
| "from numpy import bool, int, float, complex, object, unicode, str, nan, inf" |
| ) |
| NUMPY_IMPORT_NEW = """from numpy import bool, nan, inf |
| import builtins as _builtins |
| |
| int = _builtins.int |
| float = _builtins.float |
| complex = _builtins.complex |
| object = _builtins.object |
| unicode = _builtins.str |
| str = _builtins.str""" |
|
|
|
|
| def find_chumpy_dir() -> Path: |
| spec = importlib.util.find_spec("chumpy") |
| if spec is None: |
| raise SystemExit( |
| "chumpy is not installed. Run: " |
| "pip install --no-build-isolation chumpy==0.70" |
| ) |
| if spec.submodule_search_locations: |
| return Path(next(iter(spec.submodule_search_locations))) |
| if spec.origin: |
| return Path(spec.origin).parent |
| raise SystemExit("Could not locate the installed chumpy package.") |
|
|
|
|
| def patch_file(path: Path, replacements: list[tuple[str, str]]) -> bool: |
| text = path.read_text(encoding="utf-8") |
| updated = text |
| for old, new in replacements: |
| updated = updated.replace(old, new) |
| if updated == text: |
| return False |
| path.write_text(updated, encoding="utf-8") |
| return True |
|
|
|
|
| def main() -> None: |
| chumpy_dir = find_chumpy_dir() |
|
|
| changed = [] |
| ch_py = chumpy_dir / "ch.py" |
| init_py = chumpy_dir / "__init__.py" |
|
|
| if patch_file(ch_py, [(GETARGSPEC_OLD, GETARGSPEC_NEW)]): |
| changed.append(str(ch_py)) |
|
|
| if patch_file(init_py, [(NUMPY_IMPORT_OLD, NUMPY_IMPORT_NEW)]): |
| changed.append(str(init_py)) |
|
|
| if changed: |
| print("Patched chumpy files:") |
| for path in changed: |
| print(f" {path}") |
| else: |
| print("chumpy is already patched.") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
