save

BrosnanYuen · Oct 5, 2023 · cb286da · cb286da
1 parent 8fe8125
commit cb286da
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -114,3 +114,15 @@ RayBNN_DataLoader::Model::YAML::read(
 );
 ```
 
+
+
+# Write YAML Model Information
+```
+RayBNN_DataLoader::Model::YAML::write(
+    "./sample.yaml",
+
+    &modeldata_string,
+    &modeldata_float,
+    &modeldata_int,
+);
+```
diff --git a/tests/test_parse_YAML.rs b/tests/test_parse_YAML.rs
@@ -120,4 +120,7 @@ fn test_parse_YAML() {
     assert!(modeldata_float.contains_key("delete_rate"));
     assert_eq!(modeldata_float["delete_rate"].clone(), 0.1);
 
+
+
+    std::fs::remove_file("./sample.yaml");
 }