Checks

src/progpy/data_models/lstm_model.py

@@ -2,6 +2,7 @@
 # National Aeronautics and Space Administration. All Rights Reserved.
 
 from collections import abc
+import importlib.util
 from itertools import chain
 import matplotlib.pyplot as plt
 from numbers import Number
@@ -535,7 +536,10 @@ def from_data(cls, inputs, outputs, event_states=None, t_met=None, **kwargs):
  params['normalization'] = (z_mean, z_std)
 
  # Tensorflow is imported here to avoid importing it if not needed
- from tensorflow import keras
+ try:
+ from tensorflow import keras
+ except ImportError as e:
+ raise ImportError("Missing required dependencies for data-driven model. ProgPy was imported directly. Instead import with datadriven dependencies using pip3 install progpy[datadriven] or pip3 install -e '.[datadriven]' (if installing from local copy)")
 
  # Build model
  callbacks = [ ]