Fix test function

WMD-group · Aug 31, 2023 · d1b29d4 · d1b29d4
1 parent f45e748
commit d1b29d4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/bin/run_matbench_benchmark.py b/bin/run_matbench_benchmark.py
@@ -256,6 +256,9 @@ def featurize_species(X, input_type, species_dictionary, species_embeddings, ato
 
     logger.info(f"architecture: {architecture}")
     logger.info(f"val_size: {args.val_size}")
+    logger.info(f"seed: {args.seed}")
+    logger.info(f"max epochs: {args.epochs}")
+    logger.info(f"Matbench task: {args.task}")
     # TODO log other params
 
     for fold in task.folds:
@@ -327,7 +330,7 @@ def featurize_species(X, input_type, species_dictionary, species_embeddings, ato
         test_inputs = task.get_test_data(fold, include_target=False)
         if args.species:
             X_test = featurize_species(
-                test_inputs, input_type, atom_dictionary, atom_embeddings, pool
+                test_inputs, input_type,species_dictionary, species_embeddings, atom_dictionary, atom_embeddings, pool
             )
         else:
             X_test = featurize(