Fix an undefined variable corner-case.

PiperOrigin-RevId: 553265728
google-research · Aug 2, 2023 · 8d19d29 · 8d19d29
1 parent 7b69438
commit 8d19d29
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/chirp/data/utils.py b/chirp/data/utils.py
@@ -72,6 +72,7 @@ def get_dataset(
   read_config = tfds.ReadConfig(add_tfds_id=True)
 
   datasets = []
+  dataset_info = None
   for dataset_dir in dataset_directory:
     if tfds_data_dir:
       tfds.core.add_data_dir(tfds_data_dir)
@@ -106,6 +107,8 @@ def get_dataset(
         )
     )
   ds = ds.prefetch(2)
+  if dataset_info is None:
+    raise RuntimeError('No datasets loaded.')
   return ds, dataset_info