Ignore axis argument in numba CumOp when input is 1d

[jessegrabowski]

Description

Numba CumOp is passing the axis argument to np.cumsum and np.cumprod when ndim == 1, but the axis argument is not supported. Since there's only one dimension, we can just ignore it anyway, which this PR does.

Related Issue

• Closes extra_ops.cumsum raises in numba backend on 1d input when the axis argument is specified #1689
• Related to #

Checklist

• Checked that the pre-commit linting/style checks pass
• Included tests that prove the fix is effective or that the new feature works
• Added necessary documentation (docstrings and/or example notebooks)
• If you are a pro: each commit corresponds to a relevant logical change

Type of change

• New feature / enhancement
• Bug fix
• Documentation
• Maintenance
• Other (please specify):

---

📚 Documentation preview 📚: https://pytensor--1691.org.readthedocs.build/en/1691/

[Copilot]

Pull Request Overview

This PR fixes a bug in the Numba implementation of cumulative operations (CumOp) where the axis argument was incorrectly passed to NumPy's cumsum and cumprod functions for 1D inputs, causing an error since these functions don't support the axis parameter for 1D arrays in Numba.

Key changes:

• Removed the axis argument from np.cumsum and np.cumprod calls when input is 1D
• Added test coverage for 1D cumulative operations with explicit axis argument

Reviewed Changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated no comments.

File	Description
pytensor/link/numba/dispatch/extra_ops.py	Removed axis parameter from np.cumsum and np.cumprod calls in 1D case
tests/link/numba/test_extra_ops.py	Added test for 1D cumulative operations with axis=0 argument

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 81.65%. Comparing base (17c675a) to head (01e3e2c).
⚠️ Report is 2 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1691      +/-   ##
==========================================
+ Coverage   81.61%   81.65%   +0.04%     
==========================================
  Files         242      244       +2     
  Lines       53537    53590      +53     
  Branches     9433     9438       +5     
==========================================
+ Hits        43695    43760      +65     
+ Misses       7366     7350      -16     
- Partials     2476     2480       +4     

Files with missing lines	Coverage Δ
pytensor/link/numba/dispatch/extra_ops.py	96.09% <100.00%> (+3.90%)	⬆️

... and 5 files with indirect coverage changes

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.