From 01e435e840da4999b559309a9ce17bdd24d15a89 Mon Sep 17 00:00:00 2001 From: TF-Agents Team Date: Tue, 19 Dec 2023 10:39:00 -0800 Subject: [PATCH] Catch FileNotFoundError when dependency of atari_py module does not exist. PiperOrigin-RevId: 592278849 Change-Id: I7b96b9a5f57ef95a4edf790695f41a9377f8c98a --- tf_agents/environments/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tf_agents/environments/__init__.py b/tf_agents/environments/__init__.py index ff5b95d7a..381279b06 100644 --- a/tf_agents/environments/__init__.py +++ b/tf_agents/environments/__init__.py @@ -37,7 +37,7 @@ from tf_agents.environments import suite_mujoco from tf_agents.environments import suite_pybullet from tf_agents.environments.gym_wrapper import GymWrapper -except (ImportError, ModuleNotFoundError): +except (ImportError, ModuleNotFoundError, FileNotFoundError): pass from tf_agents.environments.batched_py_environment import BatchedPyEnvironment