fix: filter QC status column properly

AllenNeuralDynamics · Feb 14, 2025 · e5f157c · e5f157c
1 parent 3341f1f
commit e5f157c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/aind_qc_portal/projects/dataset.py b/src/aind_qc_portal/projects/dataset.py
@@ -179,7 +179,6 @@ def _get_assets(self):
             ),
             axis=1,
         )
-        print(self._df["QC Status"].values[0])
 
         self._df.drop(
             columns=["qc_link", "operator", "session_start_time", "location"]
@@ -215,7 +214,7 @@ def _data_filtered(self) -> pd.DataFrame:
 
         if self.status_filter != "All":
             filtered_df = filtered_df[
-                filtered_df["QC Status"] == self.status_filter
+                filtered_df["QC Status"].str.contains(self.status_filter, na=False)
             ]
 
         return filtered_df