2
\$\begingroup\$

How can I improve this code? It renders a ML env with openAI gym and matplotlib. I am new to coding so not sure if my variables or format or any lines can be improved.

def _render(self, obs):
    """Renders the environment.
    """

    ball_loc = obs[1]
    x, y = self.circle.center

    new_y = math.tan(-math.radians(obs[3])) * (ball_loc - self.target_location)

    self.circle.center = (ball_loc, 10 + new_y + 1)

    t_start = self.ax.transData
    coords = t_start.transform([self.target_location, 10])
    t = mpl.transforms.Affine2D().rotate_deg_around(coords[0], coords[1], -obs[3])
    t_end = t_start + t
    self.rect.set_transform(t_end)

    self.fig.canvas.draw()
    plt.pause(0.05)
Improve this question
\$\endgroup\$
2
  • 1
    \$\begingroup\$ Hi! Welcome to code review, Please add enough details about your code in the question body, anything that will be useful to people who'd want to review your code \$\endgroup\$
    – user228914
    Nov 2, 2020 at 9:41
  • 1
    \$\begingroup\$ We need to see more of the code - at the very least, your import statements; but ideally the entire program. \$\endgroup\$
    – Reinderien
    Nov 3, 2020 at 1:08

0

Reset to default

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge that you have read and understand our privacy policy and code of conduct.