Merge pull request #43 from TGSAI/41_metadata

Fixed incorrect parent key getting serialized in .zattrs
TGSAI · Jul 31, 2024 · a43b2b5 · a43b2b5
2 parents 9d1a512 + f5a48c7
commit a43b2b5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/mdio/variable.h b/mdio/variable.h
@@ -971,6 +971,13 @@ class Variable {
       if (isCloudStore) {
         outpath = ".zattrs";
       }
+
+      if (output_json["attributes"].contains("metadata")) {
+        auto metadata = output_json["attributes"]["metadata"];
+        output_json["attributes"].erase("metadata");
+        output_json["attributes"].merge_patch(metadata);
+      }
+
       return tensorstore::kvstore::Write(
           store.kvstore(), outpath,
           absl::Cord(output_json["attributes"].dump()));