updated apis

i4Ds · Oct 6, 2021 · 59d260c · 59d260c
1 parent 92edea6
commit 59d260c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/stixdcpy/__pycache__/quicklook.cpython-39.pyc b/stixdcpy/__pycache__/quicklook.cpython-39.pyc
diff --git a/stixdcpy/quicklook.py b/stixdcpy/quicklook.py
@@ -48,12 +48,15 @@ def peek(self, ax=None, legend_loc='upper right'):
         Parameters:
             ltc: light time correction flag. The default value is False. Do light time correction if ltc=True
         """
+        if not self.data:
+            ax.text(0.5,0.5, 'LC not available!')
+            return ax
 
         if not ax:
             _, ax = plt.subplots()
         dt = [datetime.datetime.utcfromtimestamp(t) for t in self.data['unix_time']]
         for i in range(5):
-            plt.plot(dt, self.data['light_curves'][str(i)], label=self.data['energy_bins']['names'][i])
+            ax.plot(dt, self.data['light_curves'][str(i)], label=self.data['energy_bins']['names'][i])
         dlt = self.data['DELTA_LIGHT_TIME']
         light_time_corrected = self.data['IS_LIGHT_TIME_CORRECTED']