Skip to content

Commit fa529a6

Browse files
committed
fix: get paddle_col from last row
1 parent f160e50 commit fa529a6

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

examples/OpenEnv_Tutorial.ipynb

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1021,8 +1021,10 @@
10211021
" for idx, val in enumerate(info_state):\n",
10221022
" if abs(val - 1.0) < 0.01: # Ball\n",
10231023
" ball_col = idx % grid_size\n",
1024-
" elif abs(val - 0.5) < 0.01: # Paddle\n",
1025-
" paddle_col = idx % grid_size\n",
1024+
" break\n",
1025+
"\n",
1026+
" last_row = info_state[-grid_size:]\n",
1027+
" paddle_col = last_row.index(1.0)\n",
10261028
"\n",
10271029
" if ball_col is not None and paddle_col is not None:\n",
10281030
" if paddle_col < ball_col:\n",

0 commit comments

Comments
 (0)