Comments (3)
Yes, definitely pin it.
It wouldn't be a bad idea to set up a test matrix and run the tests on jax==whatever we pin to and jax upstream.
I've seen this error before. My best guess: a new helper function got added to jax.random
; one that doesn't take shape
as an argument. We are running the scico.random
wrapper over all functions that aren't explicitly prohibited. This happened with threefry_2x32
and I fixed it in this commit: https://github.com/scici/scico/commit/3cb896d547f28862e3cc82d253ab15eaf4332cde. I'd suggest looking at the functions here and seeing if any of them are the problem:
https://github.com/google/jax/blob/573943125bc575feb16fc8eeaceb624b07c19941/jax/random.py#L109-L115
To keep this from happening in the future:
Maybe we should change to an allow_list
model and only wrap the functions that we intend to?
from scico.
It looks to me like the CI just grabs the latest jax: jax>=0.2.19
. This is bad, right? Given that the jax API is changing, I would suggest we pin SCICO to a specific version of jax.
from scico.
I see how to fix the immediate problem in random
, will handle it now.
from scico.
Related Issues (20)
- Error when applying the adjoint of a `DiagonalStack` of `TomographicProjector`
- Consistency of high-level class design
- GPU installation update for SCICO HOT 5
- Apparent error in docstring for `Operator.__init__` HOT 2
- Tests generate large number of numpy warnings HOT 4
- Test for `svmbir` interface hang on GPU device
- Document and test `input_dtype` and `output_dtype` policy
- BUG: Matrix Operator should add 'input_shape' and 'output_shape' parameters HOT 3
- Freezing elements of a BlockArray as part of optimization
- Import Error: `KeyError: 'asscalar'` HOT 15
- Open in Colab currently broken HOT 2
- Unit tests are slow in CI
- Replace `DeviceArray` with `Array` in the docs HOT 1
- `KeyError: 'asscalar'` when importing scico HOT 2
- `ADMM` initializer supports setting `f=None` HOT 1
- New sphinx theme needs to be chosen HOT 1
- Test hangs on GPU with `jaxlib` 0.4.2 HOT 3
- Vestigial function in `scico/linop/_circconv.py`
- Error in isbin.sh
- problem with cv2.remap in linop.LinearOperator HOT 3
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from scico.