Skip to content

Commit

Permalink
upgrade codespell to v2.3.0
Browse files Browse the repository at this point in the history
  • Loading branch information
Lookatator committed Sep 11, 2024
1 parent b3d302b commit 5eea7db
Show file tree
Hide file tree
Showing 4 changed files with 6 additions and 6 deletions.
2 changes: 1 addition & 1 deletion .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ repos:
- id: mypy

- repo: https://github.com/codespell-project/codespell
rev: v2.2.6
rev: v2.3.0
hooks:
- id: codespell
name: codespell
Expand Down
4 changes: 2 additions & 2 deletions examples/dcrlme.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
"\n",
"This notebook shows how to use QDax to find diverse and performing controllers in MDPs with [Descriptor-Conditioned Reinforcement Learning MAP-Elites (DCRL-ME)](https://arxiv.org/abs/2401.08632).\n",
"This algorithm extends and improves upon [Descriptor-Conditioned Gradients MAP-Elites (DCG-ME)](https://dl.acm.org/doi/abs/10.1145/3583131.3590503)\n",
"It can be run locally or on Google Colab. We recommand to use a GPU. This notebook will show:\n",
"It can be run locally or on Google Colab. We recommend to use a GPU. This notebook will show:\n",
"\n",
"- how to define the problem\n",
"- how to create the DCRL emitter\n",
Expand Down Expand Up @@ -200,7 +200,7 @@
"metadata": {},
"outputs": [],
"source": [
"# Define the fonction to play a step with the policy in the environment\n",
"# Define the function to play a step with the policy in the environment\n",
"def play_step_fn(\n",
" env_state: EnvState, policy_params: Params, random_key: RNGKey\n",
") -> Tuple[EnvState, Params, RNGKey, DCRLTransition]:\n",
Expand Down
4 changes: 2 additions & 2 deletions qdax/core/map_elites.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class MAPElites:
"""Core elements of the MAP-Elites algorithm.
Note: Although very similar to the GeneticAlgorithm, we decided to keep the
MAPElites class independant of the GeneticAlgorithm class at the moment to keep
MAPElites class independent of the GeneticAlgorithm class at the moment to keep
elements explicit.
Args:
Expand Down Expand Up @@ -65,7 +65,7 @@ def init(
Args:
genotypes: initial genotypes, pytree in which leaves
have shape (batch_size, num_features)
centroids: tesselation centroids of shape (batch_size, num_descriptors)
centroids: tessellation centroids of shape (batch_size, num_descriptors)
random_key: a random key used for stochastic operations.
Returns:
Expand Down
2 changes: 1 addition & 1 deletion tests/baselines_test/dcrlme_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,7 @@ def test_dcrlme() -> None:
fake_batch_obs = jnp.zeros(shape=(batch_size, env.observation_size))
init_params = jax.vmap(policy_network.init)(keys, fake_batch_obs)

# Define the fonction to play a step with the policy in the environment
# Define the function to play a step with the policy in the environment
def play_step_fn(
env_state: EnvState, policy_params: Params, random_key: RNGKey
) -> Tuple[EnvState, Params, RNGKey, DCRLTransition]:
Expand Down

0 comments on commit 5eea7db

Please sign in to comment.