add np_argmin/argmax to interpret_demo environment

2024-07-12 13:27:52 +02:00 · 2024-07-12 13:27:52 +02:00 · c80378063a
commit c80378063a
parent 513d30152b
1 changed files with 2 additions and 0 deletions
--- a/nac3standalone/demo/interpret_demo.py
+++ b/nac3standalone/demo/interpret_demo.py
@ -183,8 +183,10 @@ def patch(module):
    module.np_isinf = np.isinf
    module.np_min = np.min
    module.np_minimum = np.minimum
+    module.np_argmin = np.argmin
    module.np_max = np.max
    module.np_maximum = np.maximum
+    module.np_argmax = np.argmax
    module.np_sin = np.sin
    module.np_cos = np.cos
    module.np_exp = np.exp