diff --git a/graph_weather/models/graphs/ico.py b/graph_weather/models/graphs/ico.py index fd2e7a21..2b0e5d7d 100644 --- a/graph_weather/models/graphs/ico.py +++ b/graph_weather/models/graphs/ico.py @@ -318,7 +318,7 @@ def generate_icosphere_graph(resolution=1): print(f"Vertices: {np.isclose(vertex2[:12], vertex)}") exit() -def generate_icosphere_mapping(lat_lons, resolutions=(1,2,3,4,5,6,7)): +def generate_icosphere_mapping(lat_lons, resolutions=(1, 2, 3, 4, 5, 6, 7)): """ Generate mapping from lat/lon to icosphere index. @@ -348,7 +348,6 @@ def generate_icosphere_mapping(lat_lons, resolutions=(1,2,3,4,5,6,7)): # TODO Create Data object where there is a minimal amount of verticies (the overlapping ones are the same) # TODO Create mapping from the lat/lon to the icosphere nodes - return h3_mapping, h3_distances