Fixes walrus lint error

apache · May 23, 2023 · 022d554 · 022d554
1 parent d04f842
commit 022d554
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/superset/migrations/shared/migrate_viz/processors.py b/superset/migrations/shared/migrate_viz/processors.py
@@ -78,11 +78,9 @@ class MigratePivotTable(MigrateViz):
     }
 
     def _pre_action(self) -> None:
-        pivot_margins = self.data.get("pivot_margins")
-        if pivot_margins:
+        if pivot_margins := self.data.get("pivot_margins"):
             self.data["colTotals"] = pivot_margins
             self.data["rowTotals"] = pivot_margins
 
-        pandas_aggfunc = self.data.get("pandas_aggfunc")
-        if pandas_aggfunc:
+        if pandas_aggfunc := self.data.get("pandas_aggfunc"):
             self.data["pandas_aggfunc"] = self.aggregation_mapping[pandas_aggfunc]