Simplify import logic in dm_control/viewer/gui/__init__.py

alimuldal · Piotr Trochim · commit dab2bc674663 · 2018-10-19T16:42:34.000+01:00
PiperOrigin-RevId: 217473775
diff --git a/dm_control/viewer/gui/__init__.py b/dm_control/viewer/gui/__init__.py
@@ -20,12 +20,14 @@
 # pylint: disable=invalid-name
 
 RenderWindow = None
-try:
-  from dm_control.viewer.gui import glfw_gui
-  RenderWindow = glfw_gui.GlfwWindow
-except ImportError:
-  pass
-if not RenderWindow:
+
+if render.BACKEND == 'glfw':
+  try:
+    from dm_control.viewer.gui import glfw_gui
+    RenderWindow = glfw_gui.GlfwWindow
+  except ImportError:
+    pass
+else:
 
   def ErrorRenderWindow(*args, **kwargs):
     del args, kwargs
