thermal: return an empty DataFrame if no thermal trace were captured
diff --git a/cr2/thermal.py b/cr2/thermal.py
index b372962..516d6fe 100644
--- a/cr2/thermal.py
+++ b/cr2/thermal.py
@@ -70,7 +70,13 @@
if not self.data_csv:
self.parse_into_csv()
- self.data_frame = pd.read_csv(StringIO(self.data_csv)).set_index("time")
+ try:
+ self.data_frame = pd.read_csv(StringIO(self.data_csv)).set_index("time")
+ except StopIteration:
+ if not self.data_frame:
+ return pd.DataFrame()
+ raise
+
return self.data_frame
class Thermal(BaseThermal):