Catch FileNotFoundError when dependency of atari_py module does not e…

…xist. PiperOrigin-RevId: 592278849 Change-Id: I7b96b9a5f57ef95a4edf790695f41a9377f8c98a
tensorflow · Dec 19, 2023 · 01e435e · 01e435e
1 parent d185fb8
commit 01e435e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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