Ecosyste.ms: Issues

An open API service for providing issue and pull request metadata for open source projects.

GitHub / google-deepmind/chex issues and pull requests

#197 - AttributeError: module 'jax' has no attribute '_src'

Issue - State: closed - Opened by jenkspt over 2 years ago - 4 comments

#100 - Add test.sh for launching CI tests on a local machine.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#99 - Add support for the C++ pmap dispatch path.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#98 - Add support for default exception messages and types to assertions.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#97 - Add support for `jnp.bfloat16` to `asserts.assert_trees_all_close()`.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#96 - Use f-strings instead of str.format().

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#92 - Add support for `static_argnames` to `variants.with_jit`.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#91 - Add extract_patches to dm_pix, replicating tf.image.extract_patches

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago - 1 comment
Labels: cla: no

#90 - Remove restricted_inheritance option from Chex dataclasses

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#89 - Fix typo diffent->different

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#88 - Chex dataclass allow Generic types.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago - 1 comment
Labels: cla: no

#87 - Fix DevicesAssertTest broken by a change in error message

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#86 - Use `isinstance(x, {ShardedDeviceArray, DeviceArray})` instead of type(x) == y.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#85 - [JAX] Make sure that static arguments to jax.pmap are hashable in Chex.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#83 - Add `assert_trees_all_equal_dtypes` assertion.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#82 - Expose the assert_tree_shape_suffix to the public API.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#81 - Add assert_tree_shape_suffix to check whether arrays share the same suffix.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#80 - Typo in readme.

Pull Request - State: closed - Opened by lkhphuc over 3 years ago - 1 comment
Labels: cla: yes

#79 - Improve chex.fake_jit by also disabling internal jitting in functions such as jax.lax.scan.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago - 6 comments
Labels: cla: no

#77 - chex.fake_pmap converts output to ShardedDeviceArray

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#76 - Document that `fake_pmap` may change the output of pmapped functions.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago - 3 comments
Labels: cla: no

#75 - Add enable_asserts() and disable_asserts() to the API.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#74 - Release Chex v0.0.8

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#73 - Chex: asserts which only make sense on >1 tree now demand this.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#72 - Clean up asserts.py and improve CI workflow.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#71 - Error with Pydantic

Issue - State: open - Opened by kaiwenw over 3 years ago

#70 - Raise ValueError when no variants are selected.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#69 - * Exclude chex' internal frames from AssertionError tracebacks.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#68 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#67 - Allow Chex dataclasses to override "values", "keys", "get" and "items" as methods

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#66 - test

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#65 - Increase dataclass test coverage of the dm_tree API

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#64 - Specify non-pytree node dataclass fields

Issue - State: open - Opened by ethanluoyc over 3 years ago - 3 comments

#63 - Fix dataclass (un-)flatten functions. Fixes #62

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#62 - Dataclass breaks is_leaf function for jax.tree_map

Issue - State: closed - Opened by louiskirsch over 3 years ago - 1 comment

#61 - Update Chex to tolerate an upcoming change to how backends are managed in JAX.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#60 - [JAX] Change how JAX manages XLA platforms.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#59 - Format @variant test names to use only underscores and lowercase letters.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#58 - Chex: fix incorrect type annotation.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#57 - Testing mappable dataclasses.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#56 - DO NOT SUBMIT WITHOUT REMOVING THIS.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#55 - [chex] Fix a small bug in the handling of sets by assert_shape.

Pull Request - State: closed - Opened by copybara-service[bot] over 3 years ago
Labels: cla: yes

#54 - [chex] Allow `set`s of alternatives in expected shape for `assert_shape`.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago - 2 comments
Labels: cla: yes

#53 - [chex] Allow an ellipsis in the expected shape passed to `assert_shape`.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago - 8 comments
Labels: cla: yes

#52 - Chex: asserts which only make sense on >1 tree now demand this.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#51 - fake_pmap: add support for static_broadcasted_argnums

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#50 - Release Chex 0.0.7

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#49 - Add is_divisible assertion.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#48 - Disallow reserved field names.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#47 - Remove the outdated check as the required PR has been merged.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#46 - Create py.typed

Pull Request - State: closed - Opened by graingert almost 4 years ago - 2 comments
Labels: cla: yes

#45 - Remove "# Lint as: python3" directives.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#44 - Freeze by default, why not?

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#42 - [JAX] Use a hand-written Python wrapper for CompiledFunction.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#41 - Ensure LICENSE files are an exact copy of Apache LICENSE-2.0.txt [0].

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#40 - Fix @assert_max_traces usage checks.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#39 - Fix tree path formatting.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#38 - Define MANIFEST.in to include requirements.txt in pypi bundle. Increment the version.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#37 - [REQ] Conda recipe

Issue - State: closed - Opened by PhilipVinc almost 4 years ago - 2 comments

#36 - Fix PACKAGE_VER in pypi-publish.yml

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#35 - Add PyPI release workflow and increment the version.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#34 - Add .pylintrc and run pylint checks in CI workflow.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#33 - Fix typo

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago - 2 comments
Labels: cla: no

#32 - [JAX] Change Chex variants_test to use jnp.DeviceArray for isinstance() tests.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#31 - Fix chex typing annotations

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#30 - [JAX] Change Chex not to use _DeviceArray.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#29 - Parallelize pytest runs.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#28 - Fix @assert_max_traces usage checks.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#27 - Customize processing of `None`s in tree asserts and add `assert_tree_no_nones`.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#26 - Adds support for booleans to assert_type.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#25 - Add option to fake pmap with vmap while still jitting the result.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago - 3 comments
Labels: cla: no

#24 - Raise informative error for negative axes when asserting axis dimension.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago - 4 comments
Labels: cla: no

#23 - Expose XLA device types in pytypes.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#22 - without_jit=True for already jitted functions

Issue - State: open - Opened by fabiannagel almost 4 years ago

#21 - Add `assert_tree_shape_prefix` assertion.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#20 - Add a safeguard to setup.py to ensure correct package installation.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#19 - Disable testing against macos-latest to speed up CI.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#18 - Do not parse abseil flags to avoid side effects.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#17 - Update version.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#16 - Run tests under Python 3.9

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#15 - Run test on Python 3.9

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#14 - Correct and clarify docstring for `assert_equal_shape_suffix`.

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#13 - Use type() instead of __class__ in dataclass_test.py

Pull Request - State: closed - Opened by hbq1 almost 4 years ago
Labels: cla: yes

#12 - Include CI status badge in README.md

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: cla: yes

#11 - Include CI status badge in README.md

Pull Request - State: closed - Opened by copybara-service[bot] almost 4 years ago
Labels: duplicate, cla: yes

#10 - Chex dataclass throws an exception in Python 3.9

Issue - State: closed - Opened by dionhaefner about 4 years ago - 1 comment

#8 - Chex dataclass defaulting mappable_dataclass=True

Issue - State: closed - Opened by balancap about 4 years ago - 2 comments

#7 - Add .gitignore

Pull Request - State: closed - Opened by hbq1 over 4 years ago
Labels: cla: yes

#6 - Add .gitignore

Pull Request - State: closed - Opened by hbq1 over 4 years ago - 1 comment
Labels: cla: no

#5 - Add ability to check shapes with wildcards

Pull Request - State: closed - Opened by KristianHolsheimer over 4 years ago - 2 comments
Labels: cla: yes

#4 - DeprecationWarning for importing toolz

Issue - State: closed - Opened by NeilGirdhar over 4 years ago - 1 comment

#3 - Fake contexts by calling .start() not working

Issue - State: closed - Opened by sebamenabar over 4 years ago - 2 comments

#2 - Add pytypes.

Pull Request - State: closed - Opened by hbq1 over 4 years ago
Labels: cla: yes