rename torch Dataset to Local from 18

henryaddison · Mar 4, 2024 · afa9f45 · afa9f45
1 parent 893dc48
commit afa9f45
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/ml_downscaling_emulator/data.py b/src/ml_downscaling_emulator/data.py
@@ -1,4 +1,4 @@
-"""Loading UKCP18 data into PyTorch"""
+"""Loading UKCP Local data into PyTorch"""
 
 import cftime
 import numpy as np
@@ -13,7 +13,7 @@
 )
 
 
-class UKCP18Dataset(Dataset):
+class UKCPLocalDataset(Dataset):
     def __init__(self, ds, variables, target_variables, time_range):
         self.ds = ds
         self.variables = variables
@@ -96,7 +96,7 @@ def custom_collate(batch):
     time_range = None
     if include_time_inputs:
         time_range = TIME_RANGE
-    xr_dataset = UKCP18Dataset(xr_data, variables, target_variables, time_range)
+    xr_dataset = UKCPLocalDataset(xr_data, variables, target_variables, time_range)
     data_loader = DataLoader(
         xr_dataset, batch_size=batch_size, shuffle=shuffle, collate_fn=custom_collate
     )