Merge branch 'm-kovalsky/issue288'

microsoft · Nov 19, 2024 · 361c0da · 361c0da
2 parents ed1444f + 551eae3
commit 361c0da
Showing 1 changed file with 2 additions and 15 deletions.
diff --git a/src/sempy_labs/_model_bpa.py b/src/sempy_labs/_model_bpa.py
@@ -400,23 +400,10 @@ def translate_using_spark(rule_file):
         dfExport["Timestamp"] = now
         dfExport["RunId"] = runId
         dfExport["Configured By"] = configured_by
-
         dfExport["RunId"] = dfExport["RunId"].astype("int")
 
-        colName = "Capacity Name"
-        dfExport.insert(0, colName, dfExport.pop(colName))
-        colName = "Capacity Id"
-        dfExport.insert(1, colName, dfExport.pop(colName))
-        colName = "Workspace Name"
-        dfExport.insert(2, colName, dfExport.pop(colName))
-        colName = "Workspace Id"
-        dfExport.insert(3, colName, dfExport.pop(colName))
-        colName = "Dataset Name"
-        dfExport.insert(4, colName, dfExport.pop(colName))
-        colName = "Configured By"
-        dfExport.insert(5, colName, dfExport.pop(colName))
-
-        dfExport.columns = dfExport.columns.str.replace(" ", "_")
+        dfExport = dfExport[list(icons.bpa_schema.keys())]
+        dfExport["RunId"] = dfExport["RunId"].astype("int")
         schema = {
             key.replace(" ", "_"): value for key, value in icons.bpa_schema.items()
         }