Skip to content

Commit db3ce89

Browse files
Made DataConverter work with JSON files as well
1 parent 7679919 commit db3ce89

1 file changed

Lines changed: 10 additions & 1 deletion

File tree

mala/datahandling/data_converter.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from mala.targets.target import Target
1111
from mala.version import __version__ as mala_version
1212

13-
descriptor_input_types = ["espresso-out", "openpmd", "numpy"]
13+
descriptor_input_types = ["espresso-out", "openpmd", "numpy", "json"]
1414
target_input_types = [".cube", ".xsf", "openpmd", "numpy"]
1515
simulation_output_types = ["espresso-out"]
1616

@@ -550,6 +550,15 @@ def __convert_single_snapshot(
550550
snapshot["input"], **descriptor_calculation_kwargs
551551
)
552552
)
553+
elif description["input"] == "json":
554+
descriptor_calculation_kwargs["units"] = original_units["input"]
555+
descriptor_calculation_kwargs["use_fp64"] = use_fp64
556+
557+
tmp_input, local_size = (
558+
self.descriptor_calculator.calculate_from_json(
559+
snapshot["input"], **descriptor_calculation_kwargs
560+
)
561+
)
553562

554563
elif description["input"] == "openpmd":
555564
if self.parameters_full.descriptors.descriptors_contain_xyz:

0 commit comments

Comments
 (0)