Error handling for time and loc

inference.py

@@ -75,7 +75,11 @@ def read_geotiff(file_path: str):
     with rasterio.open(file_path) as src:
         img = src.read()
         meta = src.meta
-        coords = src.lnglat()
+        try:
+            coords = src.lnglat()
+        except:
+            # Cannot read coords
+            coords = None
 
     return img, meta, coords
 
@@ -137,7 +141,8 @@ def load_example(
 
         imgs.append(img)
         metas.append(meta)
-        location_coords.append(coords)
+        if coords is not None:
+            location_coords.append(coords)
 
         try:
             match = re.search(r'(\d{7,8}T\d{6})', file)
@@ -333,9 +338,9 @@ def main(
         file_paths=data_files, indices=input_indices, mean=mean, std=std
     )
 
-    if not temporal_coords and 'time' in coords_encoding:
+    if len(temporal_coords) != num_frames and 'time' in coords_encoding:
         coords_encoding.pop('time')
-    if location_coords is None and 'location' in coords_encoding:
+    if not len(location_coords) and 'location' in coords_encoding:
         coords_encoding.pop('location')
 
     # Create model and load checkpoint -------------------------------------------------------------