From c82e96e321ce43fb4deff7cad1e35991c9bc906c Mon Sep 17 00:00:00 2001
From: Koen Draijer <124170375+KMdraijer@users.noreply.github.com>
Date: Wed, 22 Oct 2025 10:58:53 +0200
Subject: [PATCH] Add forcespercent parameter to featurize function

Fix to previous fix
---
 src/aenet/featurize.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/aenet/featurize.py b/src/aenet/featurize.py
index 2b998af..33d2482 100644
--- a/src/aenet/featurize.py
+++ b/src/aenet/featurize.py
@@ -100,6 +100,7 @@ def generate_input_string(self,
                               deriv_outfile: str = 'data.deriv.train',
                               deriv_disp_all: bool = True,
                               deriv_N_max: int = 100000,
+                              forcespercent: int = None,
                               **kwargs):
 
         for arg in kwargs:
@@ -152,6 +153,10 @@ def generate_input_string(self,
 
         if debug:
             generate_in += "\nDEBUG\n"
+            
+        if forcespercent is not None:
+            generate_in += "\nFORCES\nforcespercent\t{}\n".format(forcespercent)
+
 
         generate_in += "\nFILES\n{}\n".format(len(xsf_file_paths))
         for p in xsf_file_paths: