From e2f24dca9f83b06d9ba65a1d0dd0a34553d45b95 Mon Sep 17 00:00:00 2001
From: tchaton <thomas@grid.ai>
Date: Mon, 29 Mar 2021 19:32:45 +0100
Subject: [PATCH] add _data_pipeline back

---
 flash/core/model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/flash/core/model.py b/flash/core/model.py
index a76574d180..6cc7bcda5f 100644
--- a/flash/core/model.py
+++ b/flash/core/model.py
@@ -78,6 +78,7 @@ def __init__(
         # TODO: should we save more? Bug on some regarding yaml if we save metrics
         self.save_hyperparameters("learning_rate", "optimizer")
 
+        self._data_pipeline = None
         self._preprocess = None
         self._postprocess = None