Fix a test

Lightning-Universe · May 12, 2021 · 2dc2410 · 2dc2410
1 parent 9cf62ee
commit 2dc2410
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flash_examples/predict/image_embedder.py b/flash_examples/predict/image_embedder.py
@@ -33,7 +33,7 @@
 random_image = torch.randn(1, 3, 244, 244)
 
 # 6. Generate an embedding from this random image.
-embeddings = embedder.predict(random_image, data_source="tensor")
+embeddings = embedder.predict(random_image, data_source="tensors")
 
 # 7. Print embeddings shape
 print(embeddings[0].shape)