janosh
diff --git a/‎matbench_discovery/plot_scripts/hist_classified_stable_as_func_of_hull_dist.py ‎matbench_discovery/plot_scripts/hist_classified_stable_vs_hull_dist.py b/‎matbench_discovery/plot_scripts/hist_classified_stable_as_func_of_hull_dist.py ‎matbench_discovery/plot_scripts/hist_classified_stable_vs_hull_dist.py
diff --git a/‎matbench_discovery/plot_scripts/hist_classified_stable_as_func_of_hull_dist_batches.py ‎matbench_discovery/plot_scripts/hist_classified_stable_vs_hull_dist_batches.py b/‎matbench_discovery/plot_scripts/hist_classified_stable_as_func_of_hull_dist_batches.py ‎matbench_discovery/plot_scripts/hist_classified_stable_vs_hull_dist_batches.py
diff --git a/‎models/bowsr/slurm_array_bowsr_wbm.py ‎models/bowsr/test_bowsr.py b/‎models/bowsr/slurm_array_bowsr_wbm.py ‎models/bowsr/test_bowsr.py
diff --git a/‎models/cgcnn/test_cgcnn_ensemble.py ‎models/cgcnn/test_cgcnn.py b/‎models/cgcnn/test_cgcnn_ensemble.py ‎models/cgcnn/test_cgcnn.py
diff --git a/‎models/cgcnn/slurm_train_cgcnn_ensemble.py ‎models/cgcnn/train_cgcnn.py b/‎models/cgcnn/slurm_train_cgcnn_ensemble.py ‎models/cgcnn/train_cgcnn.py
diff --git a/‎models/m3gnet/join_m3gnet_relax_results.py ‎models/m3gnet/join_m3gnet_results.py b/‎models/m3gnet/join_m3gnet_relax_results.py ‎models/m3gnet/join_m3gnet_results.py
diff --git a/‎models/m3gnet/slurm_array_m3gnet_wbm.py ‎models/m3gnet/test_m3gnet.py b/‎models/m3gnet/slurm_array_m3gnet_wbm.py ‎models/m3gnet/test_m3gnet.py
diff --git a/‎models/megnet/slurm_array_megnet_wbm.py ‎models/megnet/test_megnet.py
+4-1 b/‎models/megnet/slurm_array_megnet_wbm.py ‎models/megnet/test_megnet.py
+4-1
diff --git a/‎models/wrenformer/mp/test_wrenformer_ensemble.py ‎models/wrenformer/mp/test_wrenformer.py b/‎models/wrenformer/mp/test_wrenformer_ensemble.py ‎models/wrenformer/mp/test_wrenformer.py
diff --git a/‎models/wrenformer/slurm_train_wrenformer_ensemble.py ‎models/wrenformer/train_wrenformer.py b/‎models/wrenformer/slurm_train_wrenformer_ensemble.py ‎models/wrenformer/train_wrenformer.py
@@ -36,6 +36,7 @@
     partition="icelake-himem",
     account="LEE-SL3-CPU",
     time=(slurm_max_job_time := "12:0:0"),
+    slurm_flags=("--mem", "30G"),
     # TF_CPP_MIN_LOG_LEVEL=2 means INFO and WARNING logs are not printed
     # https://stackoverflow.com/a/40982782
     pre_cmd="TF_CPP_MIN_LOG_LEVEL=2",
@@ -89,7 +90,9 @@
     raise ValueError(f"Unknown {task_type = }")
 
 megnet_e_form_preds = {}
-for material_id, structure in tqdm(structures.items(), total=len(structures)):
+for material_id, structure in tqdm(
+    structures.items(), disable=None, total=len(structures)
+):
     if material_id in megnet_e_form_preds:
         continue
     try: