zulissimeta commited on
Commit
f7a6a0c
·
1 Parent(s): 57ae296
Files changed (1) hide show
  1. simulation_scripts.py +2 -2
simulation_scripts.py CHANGED
@@ -182,7 +182,7 @@ def run_md_simulation(
182
  atoms = load_check_ase_atoms(structure_file)
183
 
184
  # Check if the file is an example
185
- example = hash_file(structure_file) in EXAMPLE_FILE_HASHES
186
 
187
  atoms.info["charge"] = total_charge
188
  atoms.info["spin"] = spin_multiplicity
@@ -337,7 +337,7 @@ def run_relaxation_simulation(
337
  atoms = load_check_ase_atoms(structure_file)
338
 
339
  # Check if the file is an example
340
- example = hash_file(structure_file) in EXAMPLE_FILE_HASHES
341
 
342
  # Center things for consistency in visualization
343
  atoms.positions -= atoms.get_center_of_mass()
 
182
  atoms = load_check_ase_atoms(structure_file)
183
 
184
  # Check if the file is an example
185
+ example = hash_file(structure_file) in EXAMPLE_FILE_HASHES or check_catalysis_example(atoms)
186
 
187
  atoms.info["charge"] = total_charge
188
  atoms.info["spin"] = spin_multiplicity
 
337
  atoms = load_check_ase_atoms(structure_file)
338
 
339
  # Check if the file is an example
340
+ example = hash_file(structure_file) in EXAMPLE_FILE_HASHES or check_catalysis_example(atoms)
341
 
342
  # Center things for consistency in visualization
343
  atoms.positions -= atoms.get_center_of_mass()