fix: fix exported code formatting for UMAP (#139)

datamole-ai · Sep 20, 2023 · a7fafbb · a7fafbb
1 parent 693a616
commit a7fafbb
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/edvart/report_sections/umap.py b/edvart/report_sections/umap.py
@@ -189,17 +189,15 @@ def add_cells(self, cells: List[Dict[str, Any]]) -> None:
                 n_neighbors={self.n_neighbors},
                 min_dist={self.min_dist},
                 metric='{self.metric}',
-            """[
-                :-1
-            ]
+            """.rstrip()
         )
         if self.columns is not None:
-            default_call += f"    columns={self.columns},"
+            default_call += f"\n    columns={self.columns},"
         if self.color_col is not None:
-            default_call += f"    color_col='{self.color_col}',"
+            default_call += f"\n    color_col='{self.color_col}',"
         if not self.interactive:
-            default_call += "    interactive=False,"
-        default_call += ")"
+            default_call += "\n    interactive=False,"
+        default_call += "\n)"
 
         if self.verbosity <= Verbosity.MEDIUM:
             code = default_call