Skip to content

Commit

Permalink
chore: typos
Browse files Browse the repository at this point in the history
  • Loading branch information
callumtilbury committed Feb 26, 2024
1 parent 0126cb6 commit d52187c
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions examples/matrax_iql_example.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -747,7 +747,7 @@
"\n",
"\n",
"def action_step_store_scan(obs_env_mem_key_ts, _):\n",
" \"\"\"A scan-compatible verson of action_step_store.\"\"\"\n",
" \"\"\"A scan-compatible version of action_step_store.\"\"\"\n",
"\n",
" # Unpack the initial state\n",
" obs, env_state, agent_memory_state, key, agent_params, ts = obs_env_mem_key_ts\n",
Expand Down Expand Up @@ -784,7 +784,7 @@
"\n",
"\n",
"def learn_scan(learn_state, _):\n",
" \"\"\"A nested, scan-compatible verson of learn and action_step_store.\"\"\"\n",
" \"\"\"A nested, scan-compatible version of learn and action_step_store.\"\"\"\n",
"\n",
" # Unpack the initial state for learn\n",
" (\n",
Expand Down Expand Up @@ -875,7 +875,7 @@
"\n",
"\n",
"def evaluate_scan(evaluate_state, _):\n",
" \"\"\"A nested, scan-compatible verson of evaluate, learn and action_step_store.\"\"\"\n",
" \"\"\"A nested, scan-compatible version of evaluate, learn and action_step_store.\"\"\"\n",
"\n",
" # Unpack the initial state for evaluate\n",
" (\n",
Expand Down Expand Up @@ -983,7 +983,7 @@
" env_state,\n",
" ts,\n",
" )\n",
" # Perform scan over evalute, learn and action_step_store\n",
" # Perform scan over evaluate, learn and action_step_store\n",
" evaluate_state, evaluator_returns = jax.lax.scan(\n",
" evaluate_scan, evaluate_state, xs=None, length=outer\n",
" )\n",
Expand Down

0 comments on commit d52187c

Please sign in to comment.