Fix relative paths

Bench: 1736583
codedeliveryservice · Apr 16, 2024 · 335d014 · 335d014
1 parent 4c8d6d5
commit 335d014
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/build/build.rs b/build/build.rs
@@ -2,16 +2,21 @@ use std::{
     env,
     fs::File,
     io::{BufWriter, Write},
-    path::Path,
+    path::{Path, PathBuf},
 };
 
 mod attacks;
 mod magics;
 mod maps;
 
 fn main() {
-    let model = env::var("EVALFILE").unwrap_or(String::from("../networks/model.nnue"));
-    println!("cargo:rustc-env=MODEL={model}");
+    let path = env::var("EVALFILE").unwrap_or_else(|_| {
+        let path = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
+        let path = path.join("networks").join("model.nnue");
+        path.to_str().unwrap().to_string()
+    });
+
+    println!("cargo:rustc-env=MODEL={path}");
 
     let dir = env::var("OUT_DIR").unwrap();
     let path = Path::new(&dir).join("lookup.rs");