Python エラーの発生原因がわかりません

Question

前提・実現したいこと

現在迷路のゴールに向かわせるエージェントを作成しているのですが

定義をしているはずなのに
NameError: name 'next_direction' is not defined
というエラーが出てしまいます。

以下のサイトを参考に学習を進めています。
https://book.mynavi.jp/manatee/detail/id=88297

初めての質問で至らない点もあるかもしれませんがよろしくお願いします。

発生している問題・エラーメッセージ

NameError: name 'next_direction' is not defined

該当のソースコード

def get_action_and_next_s(pi, s):
    direction = ["up", "right", "down", "left"]

    next_direction = np.random.choice(direction, p=pi[s, :])
 
    if next_direction == "up":
        action = 0
        s_next = s - 3  # 上に移動するときは状態の数字が3小さくなる
    elif next_direction == "right":
        action = 1
        s_next = s + 1  # 右に移動するときは状態の数字が1大きくなる
    elif next_direction == "down":
        action = 2
        s_next = s + 3  # 下に移動するときは状態の数字が3大きくなる
    elif next_direction == "left":
        action = 3
        s_next = s - 1  # 左に移動するときは状態の数字が1小さくなる
 
    return [action, s_next]

試したこと

スペルミスなども確認しましたが見当たりませんでした。

Comments

[tiitoi]

リンク先をコピペして実行しましたが、エラーが再現しません。質問のコードで NameError: name 'next_direction' is not defined が発生するような要因も見当たりません。

[退会済みユーザー]

私も試してみましたが問題なく実行できました。
他のコードや変更前のコードを実行しているのかもしれません。

[natuorange]

コピペしても変わらずエラーが出たのですが、もう一度一番最初からやり直してみたところできました！
ありがとうございました。

[aokikenichi]

解決されたようですので、解決済みにしていただけると助かります。