From 9ef50bb835d32ade74727c1c8b12ce05ae33dc8e Mon Sep 17 00:00:00 2001
From: Mikko Kotila <mailme@mikkokotila.com>
Date: Sun, 21 Apr 2024 10:48:50 +0300
Subject: [PATCH] adding @tf.function to test function

---
 tests/commands/test_scan.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/commands/test_scan.py b/tests/commands/test_scan.py
index 6e1eae4..db07df0 100644
--- a/tests/commands/test_scan.py
+++ b/tests/commands/test_scan.py
@@ -3,6 +3,7 @@ def test_scan():
     print("\n >>> start Scan()...")
 
     import talos
+    import tensorflow as tf
 
     from tensorflow.keras.losses import binary_crossentropy
     from tensorflow.keras.optimizers.legacy import Adam
@@ -19,6 +20,7 @@ def test_scan():
          'dropout': (.05, .35, .1),
          'epochs': [50]}
 
+    @tf.function
     def iris_model(x_train, y_train, x_val, y_val, params):
 
         model = Sequential()