tom/qubed
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update tree

Browse Source
This commit is contained in:
Tom Hodson 2024-12-05 15:45:43 +00:00
parent 55249d119a
commit a998f44b93
2 changed files with 7 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tree_compresser/rust_src/tree.rs
View File

@ -72,17 +72,9 @@ impl TreeNode {
pub fn to_py_dict(&self, py: Python) -> PyResult<PyObject> {
let py_dict = PyDict::new(py);
let formatted_key = format!("{}={}", self.key.key, self.key.value);
if self.children.is_empty() {
py_dict.set_item(formatted_key, PyDict::new(py))?;
} else {
let children_dict = PyDict::new(py);
for child in &self.children {
let child_key = format!("{}={}", child.key.key, child.key.value);
children_dict.set_item(child_key, child.to_py_dict(py)?)?;
}
py_dict.set_item(formatted_key, children_dict)?;
for child in &self.children {
let child_key = format!("{}={}", child.key.key, child.key.value);
py_dict.set_item(child_key, child.to_py_dict(py)?)?;
}
Ok(py_dict.to_object(py))

6
tree_compresser/tests/open_climate_dt.py
View File

@ -1,5 +1,6 @@
from tree_traverser import backend, CompressedTree
from pathlib import Path
import json
data_path = Path("data/compressed_tree_climate_dt.json")
# Print size of file
@ -8,5 +9,6 @@ print(f"climate dt compressed tree: {data_path.stat().st_size // 1e6:.1f} MB")
print("Opening json file")
compressed_tree = CompressedTree.load(data_path)
print("Printing compressed tree")
print(compressed_tree.reconstruct_compressed_ecmwf_style())
print("Outputting compressed tree ecmwf style")
with open("data/compressed_tree_climate_dt_ecmwf_style.json", "w") as f:
json.dump(compressed_tree.reconstruct_compressed_ecmwf_style(), f)