Comments (5)
This looks to be a typeguard bug. Here's a repro without using jaxtyping at all:
from typing_extensions import Annotated
from typeguard import typechecked
@typechecked
def foo(x: Annotated[int, 'a b c']):
return 1
I'd suggest raising this as an issue on the typeguard issue tracker!
from jaxtyping.
Wasn't sure if typeguard is supposed to be able to handle "arbitrary" annotations, as it seems to only support things that are parsable by ast
. Anyway, I'm gonna open a typeguard issue to get response from the devs there, thanks :)
from jaxtyping.
(oups, wrong issue)
from jaxtyping.
FWIW I think this is now fixed with more recent typeguard releases, if anyone else would like to confirm.
from jaxtyping.
@patrick-kidger the issue has not been fixed but the issue has been closed on the typeguard repo (agronholm/typeguard#353 for anyone else running into this). My solution has been to downgrade to a pre 4.0 version of typeguard for the moment.
from jaxtyping.
Related Issues (20)
- BUG: Exception if array in args is modified in a function with the same name HOT 4
- Isinstance checks against a ShapeDtypeStruct HOT 2
- bug: can't type flax.struct.dataclass with vmapped functions HOT 3
- Disabling JAX import HOT 2
- einops-like packing notation HOT 4
- Symbolic expressions example doesn't run HOT 2
- v0.2.26 Release Failed HOT 1
- Allow two variadic shapes when it makes sense HOT 1
- Support for NestedTensors HOT 3
- `jaxtyped` Annotation fails
- jax dependency error when jax is not installed HOT 4
- Random instances / Hypothesis-like generation HOT 3
- Question: manual assertion HOT 4
- Move equinox "tree_pformat" into jaxtyping or allow users to configure their own HOT 1
- How to use with Sphinx autodoc? HOT 1
- Issues with torch.compile HOT 5
- Functions without type hints and import hook HOT 1
- Compatibility with multiprocessing / joblib - AttributeError: type object 'Float[Tensor, 'batch_size num_classes']' has no attribute 'index_variadic' HOT 4
- Can typeguard be an optional dependency? HOT 8
- Are pytorch named tensors supported, like in torchtyping? HOT 1
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 jaxtyping.