format with black

shap_e/rendering/mesh.py

@@ -89,17 +89,13 @@ class TriMesh:
 
     def write_obj(self, raw_f: BinaryIO):
         if self.has_vertex_colors():
-            vertex_colors = np.stack([self.vertex_channels[x]
+            vertex_colors = np.stack([self.vertex_channels[x] for x in "RGB"], axis=1)
-                                    for x in "RGB"], axis=1)
             vertices = [
                 "{} {} {} {} {} {}".format(*coord, *color)
                 for coord, color in zip(self.verts.tolist(), vertex_colors.tolist())
             ]
         else:
-            vertices = [
+            vertices = ["{} {} {}".format(*coord) for coord in self.verts.tolist()]
-                "{} {} {}".format(*coord)
-                for coord in self.verts.tolist()
-            ]
 
         faces = [
             "f {} {} {}".format(str(tri[0] + 1), str(tri[1] + 1), str(tri[2] + 1))