Debug for unittest and data without omitted.

AllenInstitute · Jul 18, 2023 · b7d9d06 · b7d9d06
1 parent 02999cc
commit b7d9d06
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/allensdk/brain_observatory/behavior/data_objects/stimuli/presentations.py b/allensdk/brain_observatory/behavior/data_objects/stimuli/presentations.py
@@ -525,9 +525,11 @@ def _postprocess(
         # with appending frames to the omitted flash frame log. See
         # explanation here:
         # https://github.com/AllenInstitute/AllenSDK/issues/2577
-        first_row = df.iloc[0]
-        if first_row['omitted']:
-            df = df.drop(first_row.name, axis=0)
+        if 'omitted' in df.columns:
+            first_row = df.iloc[0]
+            if not pd.isna(first_row['omitted']):
+                if first_row['omitted']:
+                    df = df.drop(first_row.name, axis=0)
         return df
 
     @staticmethod