use train colors only

liruiw · liruiw · commit bcff3964ad7c · 2023-09-25T20:15:17.000-04:00
diff --git a/cliport/utils/utils.py b/cliport/utils/utils.py
@@ -1063,12 +1063,7 @@ def get_colors(mode, n_colors=-1, **kwargs):
     return [COLORS[cn] for cn in all_color_names], all_color_names
 
 def get_colors_names(mode):
-    if mode == 'train':
-        return TRAIN_COLORS
-    elif mode == 'full':
-        return TRAIN_COLORS
-    else:
-        return TRAIN_COLORS
+    return TRAIN_COLORS
 
 def get_random_color():
     return get_colors(mode='train', n_colors=1)
