-
Notifications
You must be signed in to change notification settings - Fork 82
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Enable D*
checks
#319
Enable D*
checks
#319
Conversation
Codecov Report
📣 This organization is not using Codecov’s GitHub App Integration. We recommend you install it so Codecov can continue to function properly for your repositories. Learn more Additional details and impacted files@@ Coverage Diff @@
## main #319 +/- ##
==========================================
+ Coverage 88.78% 88.82% +0.04%
==========================================
Files 52 52
Lines 5715 5657 -58
Branches 587 863 +276
==========================================
- Hits 5074 5025 -49
+ Misses 513 506 -7
+ Partials 128 126 -2
|
@@ -292,6 +286,7 @@ def create_train_state( | |||
rng: jax.random.PRNGKeyArray, | |||
optimizer: optax.OptState, | |||
input: Union[int, Tuple[int, ...]], | |||
# TODO(michalk8): do not ignore or delete in code? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@bamos are the params
supposed to be ignored?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
params
here are for initialization and shouldn't be ignored, they're passed into NeuralTrainState.create
* Re-enable D* flake8 * Fix `D*` errors * Do not ignore D411 * Remove ignored `F401` * Enable `B008`, add `optax` to `intersphinx_mapping` * Remove old `.flake8` config * Add `SortingInitializer` to docs * Enable `E501` * Fix docs in `{costs,geometry}.py` * Update `pointcloud.py` * Fix `initializers.py` * Fix `matrix_square_root.py` * Fix `unbalanced_functions.py` * Fix more docstrings, add missing licenses * Fix typo in `FreeWassersteinBarycenter` * Fix `sinkhorn_lr.py` * Fix `models.py` * Fix `neuraldual.py` * Fix `gromov_wasserstein.py` * Fix `plot.py` * Fix `tests/` * Fix `utils.py` * Fix `k_means.py` * Fix `gaussian.py` * Fix `fit_gmm_pair.py` * Fix typo * Make `create_sample_generators` private * Unable more flake8's in docs * Add more testing flake8 * Add `SIM` * Add `RET` * Undo adding docs * Add more type hints * Use double quotes always * Add explicit quotes normalization * Udpate comment * Remove commented flake8s in `pyproject.toml` * Fix typo in typing * Skip graph factor cache test * Fix float in test instead of an integer * Improve `CostFn` docstring * Update `pairwise` docs * Remove `[d,]` from docs * [ci skip] Mention `gamma > 0` * Remove module docs, add missing copyright * [ci skip] Fix docs rendering
"
and add many additional checksjax.pure_callback
#257closes Enable and fix
D*
linter checks #315