fix docstring: session rewards are aggregated

yandexdataschool · Jun 17, 2019 · 0915fdd · 0915fdd
1 parent 2c73b58
commit 0915fdd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/week01_intro/crossentropy_method.ipynb b/week01_intro/crossentropy_method.ipynb
@@ -178,7 +178,7 @@
     "    Select states and actions from games that have rewards >= percentile\n",
     "    :param states_batch: list of lists of states, states_batch[session_i][t]\n",
     "    :param actions_batch: list of lists of actions, actions_batch[session_i][t]\n",
-    "    :param rewards_batch: list of rewards, rewards_batch[session_i][t]\n",
+    "    :param rewards_batch: list of rewards, rewards_batch[session_i]\n",
     "\n",
     "    :returns: elite_states,elite_actions, both 1D lists of states and respective actions from elite sessions\n",
     "\n",