pole.py

import gym
env = gym.make('CartPole-v0')

def ai():
    wts=[-.2,0,1,.4]
    mms=[0.76464200398541937, 2.3215749215909214, 0.20943257252883613, 2.9815988455023286]
    obs=env.reset()
    rwd=0
    for i in range(200):
        move=1 if sum([obs[i]*wts[i]/mms[i] for i in range(len(obs))]) > 0 else 0
        obs,r,d,i=env.step(move)
        rwd+=r
        if d:
            return rwd
    return rwd