added explicit return statements to feature_visualization.py functions

NREL · May 29, 2024 · 64125a8 · 64125a8
1 parent 3eeb68f
commit 64125a8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/nrel/routee/powertrain/validation/feature_visualization.py b/nrel/routee/powertrain/validation/feature_visualization.py
@@ -151,6 +151,8 @@ def visualize_features(
 
     if return_predictions:
         return predictions
+    else:
+        return None
 
 
 def contour_plot(
@@ -273,3 +275,5 @@ def contour_plot(
         plt.show()
 
     plt.close()
+
+    return None