From 950793b73e0146759796eb022b37913ad4c59888 Mon Sep 17 00:00:00 2001
From: Hannes Signer <signer@uni-potsdam.de>
Date: Thu, 11 Dec 2025 00:46:52 +0100
Subject: [PATCH] update weight path

---
 src/poet.cpp | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/poet.cpp b/src/poet.cpp
index d8077a2b9..751dcb56b 100644
--- a/src/poet.cpp
+++ b/src/poet.cpp
@@ -316,11 +316,9 @@ static Rcpp::List RunMasterLoop(RInsidePOET &R, const RuntimeParameters &params,
   if (params.ai) {
 
     if (params.function_code == 1) {
-      ai_ctx = std::make_unique<AIContext>(
-          "./bench/barite/barite_trained.weights.h5");
+      ai_ctx = std::make_unique<AIContext>("./barite_trained.weights.h5");
     } else if (params.function_code == 2) {
-      ai_ctx = std::make_unique<AIContext>(
-          "./bench/dolo/dolomite_trained.weights.h5");
+      ai_ctx = std::make_unique<AIContext>("./dolomite_trained.weights.h5");
     }
 
     R.parseEval(