fix: imports

CristianoPizzamiglio · CristianoPizzamiglio · commit 4801aeacdda4 · 2024-09-13T00:22:34.000+02:00
diff --git a/tensorflow_datasets/datasets/pneumoniamnist/pneumoniamnist_dataset_builder.py b/tensorflow_datasets/datasets/pneumoniamnist/pneumoniamnist_dataset_builder.py
@@ -1,5 +1,6 @@
 """pneumoniamnist dataset."""
 
+import tensorflow as tf
 import tensorflow_datasets.public_api as tfds
 import numpy as np
 
