fixing black error

UrbanSystemsLab · Dec 2, 2024 · aee4ef8 · aee4ef8
1 parent 483a59a
commit aee4ef8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/usl_models/notebooks/train_atmo_model.ipynb b/usl_models/notebooks/train_atmo_model.ipynb
@@ -116,7 +116,7 @@
     "    label_bucket_name=label_bucket_name,\n",
     "    sim_names=sim_names,\n",
     "    timesteps_per_day=time_steps_per_day,\n",
-    "    max_blobs=30\n",
+    "    max_blobs=30,\n",
     ")\n",
     "ds = ds.batch(batch_size=4)\n",
     "train_dataset, val_dataset, test_dataset = dataset.split_dataset(\n",

diff --git a/usl_models/usl_models/atmo_ml/dataset.py b/usl_models/usl_models/atmo_ml/dataset.py
@@ -78,8 +78,7 @@ def data_generator():
                 max_blobs=max_blobs,
             )
             label_data = load_folder_from_cloud(
-                label_bucket_name, label_folder, storage_client,
-                max_blobs=max_blobs
+                label_bucket_name, label_folder, storage_client, max_blobs=max_blobs
             )
 
             # Iterate through each spatiotemporal and label file
@@ -157,7 +156,7 @@ def load_prediction_dataset(
     timesteps_per_day: int,
     batch_size: int = 4,
     storage_client: storage.Client = None,
-    max_blobs: int | None = None
+    max_blobs: int | None = None,
 ) -> tf.data.Dataset:
 
     # Load spatial, LU index, and spatiotemporal data from their respective folders