flax
Fixes a bug in type annotations for scope.param (unbox=True should accept callable[..., T | AxisMEtadata[T]] and return T, while unbox=False should always return the same thing as what callable returning.
#4727

Merged

Fixes a bug in type annotations for scope.param (unbox=True should accept callable[..., T | AxisMEtadata[T]] and return T, while unbox=False should always return the same thing as what callable returning. #4727

copybara-service merged 1 commit into main from test_750577412

copybara-service force pushed from c8c2dbac to 1ceddda3 1 year ago

copybara-service force pushed from 1ceddda3 to 84d96f76 1 year ago

copybara-service force pushed from 84d96f76 to 9aa2cccd 1 year ago

copybara-service force pushed from 9aa2cccd to 750eefe8 1 year ago

copybara-service changed the title Fixes bug in type annotations for scope.param (unbox=True should accept callable[..., T | AxisMEtadata[T]] and return T, while unbox=False should always return the same thing as what callable returning. Fixes a bug in type annotations for scope.param (unbox=True should accept callable[..., T | AxisMEtadata[T]] and return T, while unbox=False should always return the same thing as what callable returning. 1 year ago

copybara-service force pushed from 750eefe8 to 621c9221 1 year ago

copybara-service force pushed from 621c9221 to bfff9e72 1 year ago

copybara-service force pushed from bfff9e72 to 3e8087dd 1 year ago

copybara-service force pushed from 3e8087dd to 7ba5e27d 1 year ago

copybara-service force pushed from 7ba5e27d to 00e25ea8 1 year ago

copybara-service force pushed from 00e25ea8 to 50d9ce82 1 year ago

copybara-service force pushed from 50d9ce82 to 1ccbd7d3 1 year ago

copybara-service force pushed from 1ccbd7d3 to 1e82966c 1 year ago

copybara-service force pushed from 1e82966c to e68f0519 1 year ago

Fixes a bug in type annotations for scope.param (unbox=True should ac…

54fad0a8

copybara-service force pushed from e68f0519 to 54fad0a8 1 year ago

copybara-service merged 54fad0a8 into main 1 year ago

copybara-service deleted the test_750577412 branch 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

flax Fixes a bug in type annotations for scope.param (unbox=True should accept callable[..., T | AxisMEtadata[T]] and return T, while unbox=False should always return the same thing as what callable returning. #4727 Merged

Fixes a bug in type annotations for scope.param (unbox=True should accept callable[..., T | AxisMEtadata[T]] and return T, while unbox=False should always return the same thing as what callable returning. #4727

flax
Fixes a bug in type annotations for scope.param (unbox=True should accept callable[..., T | AxisMEtadata[T]] and return T, while unbox=False should always return the same thing as what callable returning.
#4727

Merged