Update helpers/functional.py

helpers/functional.py

@@ -2,6 +2,7 @@ from fsspec.parquet import open_parquet_file
 import fsspec
 import pyarrow.parquet as pq
 from .grid import *
+from .ReadAlphaEarth import *
 import pandas as pd
 from io import BytesIO
 import os
@@ -139,6 +140,10 @@ def text_to_image(text, return_centre=False, return_gridcell=False, return_times
 
 def cell_to_image(cell, return_centre=False, return_gridcell=False, return_timestamp=False, source='Sentinel-2 L2A'):
 
+    # AEF is special - it's read live
+    if source == 'AlphaEarth Embeddings':
+        return cell_to_aef(cell, return_centre=return_centre, return_gridcell=return_gridcell, return_timestamp=return_timestamp)
+
     # 3. translate major-tom cell to row in parquet
     df = df_dict[source]
     row = cell2row(cell, df, return_row = True)