fixed missing protein columns #253

wilhelm-lab · Aug 4, 2024 · 6d66789 · 6d66789
1 parent 66e0f36
commit 6d66789
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/oktoberfest/preprocessing/preprocessing.py b/oktoberfest/preprocessing/preprocessing.py
@@ -38,8 +38,11 @@ def gen_lib(input_file: Union[str, Path]) -> Spectra:
     """
     library_df = csv.read_file(input_file)
     library_df.columns = library_df.columns.str.upper()
+    if "PROTEINS" not in library_df.obs.columns:
+        library_df.obs["PROTEINS"] = "unknown"
     var_df = Spectra._gen_vars_df()
     spec = Spectra(obs=library_df, var=var_df)
+
     spec.var_names = var_df.index
     return spec