Fixes

Lightning-Universe · Jun 18, 2021 · 7aa3835 · 7aa3835
1 parent f8bb543
commit 7aa3835
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flash/core/model.py b/flash/core/model.py
@@ -391,7 +391,7 @@ def build_data_pipeline(
             else:
                 data_source = preprocess.data_source_of_name(data_source)
 
-        if type(deserializer) == Deserializer:
+        if deserializer is None or type(deserializer) == Deserializer:
             deserializer = getattr(preprocess, "deserializer", deserializer)
 
         data_pipeline = DataPipeline(data_source, preprocess, postprocess, deserializer, serializer)