Update reinforce_cartpole.py

reinforce_cartpole.py

+import gymnasium as gym
+import torch
+import torch.nn as nn
+import torch.optim as optim
+import torch.nn.functional as F
+from torch.distributions import Categorical
+import matplotlib.pyplot as plt
+
+
+
+
+
+
+
+
+
+
 # Create the environment
 env = gym.make("CartPole-v1", render_mode="human")