fix CaptionedMetricsTable proprietary models toggle

Author: janosh

matbench_discovery/preds.py

@@ -61,7 +61,7 @@ class PredFiles(Files):
     # alignn_pretrained = "alignn/2023-06-03-mp-e-form-alignn-wbm-IS2RE.csv.gz"
     # alignn_ff = "alignn_ff/2023-07-11-alignn-ff-wbm-IS2RE.csv.gz"
 
-    # gnome = "gnome/2023-11-01-gnome-preds-50076332.csv.gz"
+    gnome = "gnome/2023-11-01-gnome-preds-50076332.csv.gz"
 
 
 # key_map maps model keys to pretty labels

scripts/model_figs/make_metrics_tables.py

@@ -14,8 +14,9 @@
 from pymatviz.utils import si_fmt
 from sklearn.dummy import DummyClassifier
 
-from matbench_discovery import PDF_FIGS, SCRIPTS, SITE_FIGS, Key, Open
+from matbench_discovery import PDF_FIGS, SCRIPTS, SITE_FIGS
 from matbench_discovery.data import DATA_FILES, df_wbm
+from matbench_discovery.enums import Key, Model, Open
 from matbench_discovery.metrics import stable_metrics
 from matbench_discovery.models import MODEL_METADATA
 from matbench_discovery.preds import df_metrics, df_metrics_10k, df_metrics_uniq_protos
@@ -159,9 +160,11 @@
         lower_is_better, " ↓"
     )
     styler.relabel_index(
-        [f"{col}{arrow_suffix.get(col, '')}" for col in df_table],
-        axis="columns",
-    )
+        [f"{col}{arrow_suffix.get(col, '')}" for col in df_table], axis="columns"
+    ).set_uuid("")
+
+    # add CSS class 'proprietary' to cells of proprietary models (openness != OSOD)
+    styler.set_td_classes(df_table.T.assign(GNoME="proprietary")[[Model.gnome]].T)
 
     # export model metrics as styled HTML table and Svelte component
     # get index of MAE column

scripts/model_figs/run_all.py

@@ -22,7 +22,7 @@
 go.Figure.show = lambda *_args, **_kwargs: None
 Dash.run = lambda *_args, **_kwargs: None
 
-# subtract __file__ to avoid this file calling itself
+# subtract __file__ to prevent this file from calling itself
 scripts = set(glob(f"{module_dir}/*.py")) - {__file__}