diff --git a/mhcflurry/class1_ligandome_predictor.py b/mhcflurry/class1_ligandome_predictor.py
index cc851e6994284b676a2bab216f0f5c2f42aae21e..e96e7124b64d49be2c64d16075f437063c977862 100644
--- a/mhcflurry/class1_ligandome_predictor.py
+++ b/mhcflurry/class1_ligandome_predictor.py
@@ -625,6 +625,13 @@ class Class1LigandomePredictor(object):
         fit_info["num_points"] = len(labels)
         self.fit_info.append(dict(fit_info))
 
+        return {
+            'batch_generator': batch_generator,
+            'last_x': x_dict_with_random_negatives,
+            'last_y': [encoded_y1, encoded_y2, encoded_y2],
+            'fit_info': fit_info,
+        }
+
     def predict(
             self,
             peptides,