Skip to content

Commit

Permalink
executorlib update.
Browse files Browse the repository at this point in the history
  • Loading branch information
mgt16-LANL committed Dec 17, 2024
1 parent 6ec7be6 commit a76cc1a
Showing 1 changed file with 9 additions and 7 deletions.
16 changes: 9 additions & 7 deletions development/solubility_dev/workflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -147,7 +147,8 @@ def gen_confs(smi):
except:
results = dict()
results["uff_energy"] = float(
conf.split("\n")[1].split("=")[1].split()[0])
conf.split("\n")[1].split("=")[1].split()[0]
)
results["uff_mol2"] = conf
results["smiles"] = insmiles
results["xtb_solvent"] = solvent
Expand All @@ -161,18 +162,19 @@ def encode_smi(insmi):

# Log output to file from subprocess
out_df = pd.DataFrame(results_list)
out_df.to_pickle(savedir_p / ("Smiles_{}.pkl".format(encode_smi(insmiles)))
)
out_df.to_pickle(savedir_p / ("Smiles_{}.pkl".format(encode_smi(insmiles))))
return results_list


with flux.job.FluxExecutor() as flux_exe:
with Executor(
max_workers=max_workers, # total number of cores available to the Executor
cores_per_worker=1,
threads_per_core=1,
cwd=cwd,
openmpi_oversubscribe=False, # not available with flux
resource_dict={
"cores": 1,
"threads_per_core": 1,
"cwd": cwd,
"openmpi_oversubscribe": False,
},
flux_executor=flux_exe,
block_allocation=True, # reuse existing processes with fixed resources
) as exe:
Expand Down

0 comments on commit a76cc1a

Please sign in to comment.