fix import naming and organization

Ekhao · Ekhao · commit b807cfe4d5ab · 2024-08-08T09:51:17.000+02:00
diff --git a/tensorflow_datasets/image_classification/wake_vision/wake_vision_dataset_builder.py b/tensorflow_datasets/image_classification/wake_vision/wake_vision_dataset_builder.py
@@ -1,8 +1,9 @@
 """wake_vision dataset."""
 
-import tensorflow_datasets.public_api as tfds
 import os
 
+import tensorflow_datasets as tfds
+
 _TRAIN_IMAGE_IDS = [
   9270406, 9270356, 9270408, 9270367, 9270349, 9270351, 9270390, 9270375,
   9270387, 9270370, 9270396, 9270340, 9270411, 9270369, 9270357, 9270378,
