Fix bug, compare against enum

Xilinx · Nov 10, 2021 · e2f4455 · e2f4455
1 parent a4b0262
commit e2f4455
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pynq_composable/video.py b/pynq_composable/video.py
@@ -162,7 +162,7 @@ def __init__(self, ol: Overlay, source: VSource = VSource.HDMI) -> None:
         self._thread = threading.Lock()
         self._running = None
 
-        if self._source == 'HDMI':
+        if self._source == VSource.HDMI:
             self._source_in = ol.video.hdmi_in
             self._source_in.frontend.set_phy(ol.video.phy.vid_phy_controller)
         else:
@@ -171,7 +171,7 @@ def __init__(self, ol: Overlay, source: VSource = VSource.HDMI) -> None:
     def start(self):
         """Configure and start the HDMI"""
         if not self._started:
-            if self._source == 'HDMI':
+            if self._source == VSource.HDMI:
                 self._source_in.configure()
                 videomode = self._source_in.mode
             else: