Update app.py

app.py

@@ -67,7 +67,7 @@ def data_comparison(df):
         #data_kmeans['distance_from_centroid'] = data_kmeans.apply(distance_from_centroid, axis=1)
 
     selection = alt.selection_multi(fields=['cluster','label'])
-    color = alt.condition(alt.datum.slice == 'high-loss', alt.Color('cluster:N', scale = alt.Scale(domain=df.cluster.tolist())), alt.value("lightgray"))
+    color = alt.condition(alt.datum.slice == 'high-loss', alt.Color('cluster:N', scale = alt.Scale(domain=df.cluster.unique().tolist())), alt.value("lightgray"))
     # color = alt.condition(selection, 
     #                        alt.Color('cluster:Q', legend=None),
     #                         # scale = alt.Scale(domain = pop_domain,range=color_range)),
@@ -88,7 +88,7 @@ def data_comparison(df):
     ).interactive()
 
     legend = alt.Chart(df).mark_point().encode(
-        y=alt.Y('cluster:O', axis=alt.Axis(orient='right'), title=""),
+        y=alt.Y('cluster:N', axis=alt.Axis(orient='right'), title=""),
         x=alt.X("label"),
         shape=alt.Shape('label', scale=alt.Scale(
         range=['circle', 'diamond']), legend=None),