Skip to content

Commit 3bb23ab

Browse files
committed
feat: use mesh dir and dynamic dir from metadata
1 parent 53b8e41 commit 3bb23ab

File tree

1 file changed

+6
-12
lines changed

1 file changed

+6
-12
lines changed

workers/mesh_worker.py

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -37,9 +37,12 @@ def callback(payload):
3737
)
3838

3939
try:
40-
mesh_dir = cg.meta.dataset_info["mesh"]
41-
mesh_meta = cg.meta.dataset_info["mesh_metadata"]
42-
cv_unsharded_mesh_dir = mesh_meta.get("unsharded_mesh_dir", f"dynamic_{cg.graph_id}")
40+
mesh_meta = cg.meta.custom_data["mesh"]
41+
mesh_dir = mesh_meta["dir"]
42+
layer = mesh_meta["max_layer"]
43+
mip = mesh_meta["mip"]
44+
err = mesh_meta["max_error"]
45+
cv_unsharded_mesh_dir = mesh_meta.get("dynamic_mesh_dir", "dynamic")
4346
except KeyError:
4447
logging.warning(f"No metadata found for {cg.graph_id}; ignoring...")
4548
return
@@ -48,15 +51,6 @@ def callback(payload):
4851
cg.meta.data_source.WATERSHED, mesh_dir, cv_unsharded_mesh_dir
4952
)
5053

51-
try:
52-
mesh_data = cg.meta.custom_data["mesh"]
53-
layer = mesh_data["max_layer"]
54-
mip = mesh_data["mip"]
55-
err = mesh_data["max_error"]
56-
except KeyError:
57-
return
58-
59-
6054
logging.log(INFO_HIGH, f"remeshing {l2ids}; graph {table_id} operation {op_id}.")
6155
meshgen.remeshing(
6256
cg,

0 commit comments

Comments
 (0)