Merge pull request NVIDIA-AI-IOT#367 from bezineb5/patch-1

Missing 'os' import
jaybdub · Jan 17, 2021 · eb9eb05 · eb9eb05
2 parents ef97ba9 + 87b6928
commit eb9eb05
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/jetbot/camera/__init__.py b/jetbot/camera/__init__.py
@@ -1,12 +1,10 @@
-try:
-    DEFAULT_CAMERA = os.environ['JETBOT_DEFAULT_CAMERA']
-except:
-    DEFAULT_CAMERA = 'opencv_gst_camera'
+import os
 
+DEFAULT_CAMERA = os.environ.get('JETBOT_DEFAULT_CAMERA', 'opencv_gst_camera')
 
 if DEFAULT_CAMERA == 'zmq_camera':
     from .zmq_camera import ZmqCamera
     Camera = ZmqCamera
 else:
     from .opencv_gst_camera import OpenCvGstCamera
-    Camera = OpenCvGstCamera
+    Camera = OpenCvGstCamera