[FIX](handlers) add return value to select engine

wbenbihi · Aug 25, 2022 · e6328ab · e6328ab
1 parent a91f5f3
commit e6328ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hourglass_tensorflow/handlers/dataset.py b/hourglass_tensorflow/handlers/dataset.py
@@ -71,7 +71,7 @@ def heatmap(self) -> HTFDatasetHeatmap:
 
     def select_engine(self, data: Any) -> HTFEngine:
         try:
-            self.engine = self._engines[type(data)](metadata=self._metadata)
+            return self._engines[type(data)](metadata=self._metadata)
         except KeyError:
             raise KeyError(f"No engine available for type {type(data)}")