Fix typo (#220)

Lightning-Universe · Apr 17, 2021 · 4e68cc0 · 4e68cc0
1 parent 33c3e2d
commit 4e68cc0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flash/data/data_pipeline.py b/flash/data/data_pipeline.py
@@ -56,7 +56,7 @@ class CustomPostprocess(Postprocess):
     """
 
     PREPROCESS_FUNCS: Set[str] = _PREPROCESS_FUNCS
-    POSTPROCES_FUNCS: Set[str] = _POSTPROCESS_FUNCS
+    POSTPROCESS_FUNCS: Set[str] = _POSTPROCESS_FUNCS
 
     def __init__(self, preprocess: Optional[Preprocess] = None, postprocess: Optional[Postprocess] = None) -> None:
         self._preprocess_pipeline = preprocess or Preprocess()
@@ -344,7 +344,7 @@ def _create_uncollate_postprocessors(self, stage: RunningStage) -> _PostProcesso
 
         func_names: Dict[str, str] = {
             k: self._resolve_function_hierarchy(k, postprocess, stage, object_type=Postprocess)
-            for k in self.POSTPROCES_FUNCS
+            for k in self.POSTPROCESS_FUNCS
         }
 
         # since postprocessing is exclusive for prediction, we don't have to check the resolution hierarchy here.