diff --git a/rl_coach/agents/human_agent.py b/rl_coach/agents/human_agent.py index 2b92058..5d91ba1 100644 --- a/rl_coach/agents/human_agent.py +++ b/rl_coach/agents/human_agent.py @@ -18,7 +18,9 @@ import os from collections import OrderedDict from typing import Union -import pygame +import contextlib +with contextlib.redirect_stdout(None): + import pygame from pandas import to_pickle from rl_coach.agents.agent import Agent diff --git a/rl_coach/renderer.py b/rl_coach/renderer.py index 91a3c7a..03a13ff 100644 --- a/rl_coach/renderer.py +++ b/rl_coach/renderer.py @@ -15,7 +15,9 @@ # import numpy as np -import pygame +import contextlib +with contextlib.redirect_stdout(None): + import pygame from pygame.locals import HWSURFACE, DOUBLEBUF