1conda env create -n mmgp_tensile2d -f https://huggingface.co/fabiencasenave/mmgp_tensile2d/resolve/main/environment.yml
2conda activate mmgp_tensile2d
3pip install git+https://huggingface.co/fabiencasenave/mmgp_tensile2d
1from datasets import load_dataset
2from plaid.bridges.huggingface_bridge import huggingface_dataset_to_plaid
3import mmgp_tensile2d
4
5model = mmgp_tensile2d.load()
6
7hf_dataset = load_dataset("PLAID-datasets/Tensile2d", split="all_samples")
8ids_test = hf_dataset.description["split"]['test'][:5]
9
10dataset_test, _ = huggingface_dataset_to_plaid(hf_dataset, ids = ids_test, processes_number = 5, verbose = True)
11
12print("Check that 'U1' field is not present in test dataset: dataset_test[0].get_field('U1') =", dataset_test[0].get_field('U1'))
13
14print("Run prediction...")
15dataset_pred = model.predict(dataset_test)
16
17print("Check that 'U1' field is now present in pred dataset: dataset_pred[0].get_field('U1') =", dataset_pred[0].get_field('U1'))