tf run eagerly to get accurate pytest-cov report on solar model

NREL · Jun 27, 2024 · ebb154c · ebb154c
1 parent aca9fc7
commit ebb154c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tests/training/test_train_gan_dc.py b/tests/training/test_train_gan_dc.py
@@ -11,9 +11,7 @@
     DataHandlerH5,
 )
 from sup3r.utilities.loss_metrics import MmdMseLoss
-from sup3r.utilities.pytest.helpers import (
-    BatchHandlerTesterDC,
-)
+from sup3r.utilities.pytest.helpers import BatchHandlerTesterDC
 
 TARGET_COORD = (39.01, -105.15)
 FEATURES = ['u_100m', 'v_100m']

diff --git a/tests/training/test_train_solar.py b/tests/training/test_train_solar.py
@@ -6,6 +6,7 @@
 
 import numpy as np
 import pytest
+import tensorflow as tf
 from tensorflow.keras.losses import MeanAbsoluteError
 
 from sup3r import CONFIG_DIR
@@ -16,6 +17,9 @@
 FEATURES_S = ['clearsky_ratio', 'ghi', 'clearsky_ghi']
 TARGET_S = (39.01, -105.13)
 
+# added to get accurate pytest-cov report on tf.function
+tf.config.run_functions_eagerly(True)
+
 
 def test_solar_cc_model():
     """Test the solar climate change nsrdb super res model.