update

Lightning-Universe · May 12, 2021 · 1da7d89 · 1da7d89
1 parent c512f75
commit 1da7d89
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/flash/data/batch.py b/flash/data/batch.py
@@ -166,10 +166,10 @@ def forward(self, samples: Sequence[Any]) -> Any:
                 samples = type(_samples)(_samples)
 
                 with self._collate_context:
-                    samples, metada = self._extract_metadata(samples)
+                    samples, metadata = self._extract_metadata(samples)
                     samples = self.collate_fn(samples)
-                    if metada:
-                        samples[DefaultDataKeys.METADATA] = metada
+                    if metadata:
+                        samples[DefaultDataKeys.METADATA] = metadata
                     self.callback.on_collate(samples, self.stage)
 
             with self._per_batch_transform_context: