Use 'epochs' as determiner for model history presence

Author: TheCSGuy25

autoemulate/core/plotting.py

@@ -642,13 +642,14 @@ def plot_loss(
     """
 
     try:
-        history = model.loss_history
-    except:
-        raise AttributeError("Emulator does not have a Loss history")
+        has_epochs = model.epochs
+    except Exception:
+        raise AttributeError
 
     if figsize is None:
         figsize = (6, 6)
 
+    history = model.loss_history
     fig, ax = plt.subplots(figsize=figsize)
     ax.plot(range(1, len(history) + 1), history)
     ax.set_xlabel("Epochs")

tests/core/test_plotting.py

@@ -89,7 +89,7 @@ def test_plot_loss(model_class, should_raise, title):
 
     if should_raise:
         with pytest.raises(AttributeError):
-            plotting.plot_loss(model=model, title=title)
+            fig, ax = plotting.plot_loss(model=model, title=title)
         return
 
     fig, ax = plotting.plot_loss(model=model, title=title)