Adapt to removal of statistics state_unit_of_measurement (#13935)

* Adapt to removal of statistics state_unit_of_measurement

* Tweak

src/components/chart/statistics-chart.ts

@@ -21,6 +21,7 @@ import {
   numberFormatToLocale,
 } from "../../common/number/format_number";
 import {
+  getDisplayUnit,
   getStatisticLabel,
   getStatisticMetadata,
   Statistics,
@@ -258,8 +259,11 @@ class StatisticsChart extends LitElement {
 
       if (!this.unit) {
         if (unit === undefined) {
-          unit = meta?.state_unit_of_measurement;
+          unit = getDisplayUnit(this.hass, firstStat.statistic_id, meta);
-        } else if (unit !== meta?.state_unit_of_measurement) {
+        } else if (
+          unit !== getDisplayUnit(this.hass, firstStat.statistic_id, meta)
+        ) {
+          // Clear unit if not all statistics have same unit
           unit = null;
         }
       }

src/components/entity/ha-statistic-picker.ts

@@ -5,9 +5,12 @@ import { customElement, property, query, state } from "lit/decorators";
 import memoizeOne from "memoize-one";
 import { ensureArray } from "../../common/ensure-array";
 import { fireEvent } from "../../common/dom/fire_event";
-import { computeStateName } from "../../common/entity/compute_state_name";
 import { stringCompare } from "../../common/string/compare";
-import { getStatisticIds, StatisticsMetaData } from "../../data/recorder";
+import {
+  getStatisticIds,
+  getStatisticLabel,
+  StatisticsMetaData,
+} from "../../data/recorder";
 import { PolymerChangedEvent } from "../../polymer-types";
 import { HomeAssistant } from "../../types";
 import { documentationUrl } from "../../util/documentation-url";
@@ -43,18 +46,11 @@ export class HaStatisticPicker extends LitElement {
   public includeStatisticsUnitOfMeasurement?: string | string[];
 
   /**
-   * Show only statistics displayed with these units of measurements.
+   * Show only statistics with these unit classes.
-   * @attr include-display-unit-of-measurement
+   * @attr include-unit-class
    */
-  @property({ attribute: "include-display-unit-of-measurement" })
+  @property({ attribute: "include-unit-class" })
-  public includeDisplayUnitOfMeasurement?: string | string[];
+  public includeUnitClass?: string | string[];
-
-  /**
-   * Show only statistics with these device classes.
-   * @attr include-device-classes
-   */
-  @property({ attribute: "include-device-classes" })
-  public includeDeviceClasses?: string[];
 
   /**
    * Show only statistics on entities.
@@ -97,8 +93,7 @@ export class HaStatisticPicker extends LitElement {
     (
       statisticIds: StatisticsMetaData[],
       includeStatisticsUnitOfMeasurement?: string | string[],
-      includeDisplayUnitOfMeasurement?: string | string[],
+      includeUnitClass?: string | string[],
-      includeDeviceClasses?: string[],
       entitiesOnly?: boolean
     ): Array<{ id: string; name: string; state?: HassEntity }> => {
       if (!statisticIds.length) {
@@ -113,15 +108,18 @@ export class HaStatisticPicker extends LitElement {
       }
 
       if (includeStatisticsUnitOfMeasurement) {
-        const includeUnits = ensureArray(includeStatisticsUnitOfMeasurement);
+        const includeUnits: (string | null)[] = ensureArray(
+          includeStatisticsUnitOfMeasurement
+        );
         statisticIds = statisticIds.filter((meta) =>
           includeUnits.includes(meta.statistics_unit_of_measurement)
         );
       }
-      if (includeDisplayUnitOfMeasurement) {
+      if (includeUnitClass) {
-        const includeUnits = ensureArray(includeDisplayUnitOfMeasurement);
+        const includeUnitClasses: (string | null)[] =
+          ensureArray(includeUnitClass);
         statisticIds = statisticIds.filter((meta) =>
-          includeUnits.includes(meta.state_unit_of_measurement)
+          includeUnitClasses.includes(meta.unit_class)
         );
       }
 
@@ -136,23 +134,16 @@ export class HaStatisticPicker extends LitElement {
           if (!entitiesOnly) {
             output.push({
               id: meta.statistic_id,
-              name: meta.name || meta.statistic_id,
+              name: getStatisticLabel(this.hass, meta.statistic_id, meta),
             });
           }
           return;
         }
-        if (
+        output.push({
-          !includeDeviceClasses ||
+          id: meta.statistic_id,
-          includeDeviceClasses.includes(
+          name: getStatisticLabel(this.hass, meta.statistic_id, meta),
-            entityState!.attributes.device_class || ""
+          state: entityState,
-          )
+        });
-        ) {
-          output.push({
-            id: meta.statistic_id,
-            name: computeStateName(entityState),
-            state: entityState,
-          });
-        }
       });
 
       if (!output.length) {
@@ -203,8 +194,7 @@ export class HaStatisticPicker extends LitElement {
         (this.comboBox as any).items = this._getStatistics(
           this.statisticIds!,
           this.includeStatisticsUnitOfMeasurement,
-          this.includeDisplayUnitOfMeasurement,
+          this.includeUnitClass,
-          this.includeDeviceClasses,
           this.entitiesOnly
         );
       } else {
@@ -212,8 +202,7 @@ export class HaStatisticPicker extends LitElement {
           (this.comboBox as any).items = this._getStatistics(
             this.statisticIds!,
             this.includeStatisticsUnitOfMeasurement,
-            this.includeDisplayUnitOfMeasurement,
+            this.includeUnitClass,
-            this.includeDeviceClasses,
             this.entitiesOnly
           );
         });

src/components/entity/ha-statistics-picker.ts

@@ -32,11 +32,11 @@ class HaStatisticsPicker extends LitElement {
   public includeStatisticsUnitOfMeasurement?: string[] | string;
 
   /**
-   * Show only statistics displayed with these units of measurements.
+   * Show only statistics with these unit classes.
-   * @attr include-display-unit-of-measurement
+   * @attr include-unit-class
    */
-  @property({ attribute: "include-display-unit-of-measurement" })
+  @property({ attribute: "include-unit-class" })
-  public includeDisplayUnitOfMeasurement?: string[] | string;
+  public includeUnitClass?: string | string[];
 
   /**
    * Ignore filtering of statistics type and units when only a single statistic is selected.
@@ -58,12 +58,12 @@ class HaStatisticsPicker extends LitElement {
       this.ignoreRestrictionsOnFirstStatistic &&
       this._currentStatistics.length <= 1;
 
-    const includeDisplayUnitCurrent = ignoreRestriction
-      ? undefined
-      : this.includeDisplayUnitOfMeasurement;
     const includeStatisticsUnitCurrent = ignoreRestriction
       ? undefined
       : this.includeStatisticsUnitOfMeasurement;
+    const includeUnitClassCurrent = ignoreRestriction
+      ? undefined
+      : this.includeUnitClass;
     const includeStatisticTypesCurrent = ignoreRestriction
       ? undefined
       : this.statisticTypes;
@@ -75,8 +75,8 @@ class HaStatisticsPicker extends LitElement {
             <ha-statistic-picker
               .curValue=${statisticId}
               .hass=${this.hass}
-              .includeDisplayUnitOfMeasurement=${includeDisplayUnitCurrent}
               .includeStatisticsUnitOfMeasurement=${includeStatisticsUnitCurrent}
+              .includeUnitClass=${includeUnitClassCurrent}
               .value=${statisticId}
               .statisticTypes=${includeStatisticTypesCurrent}
               .statisticIds=${this.statisticIds}
@@ -89,10 +89,9 @@ class HaStatisticsPicker extends LitElement {
       <div>
         <ha-statistic-picker
           .hass=${this.hass}
-          .includeDisplayUnitOfMeasurement=${this
-            .includeDisplayUnitOfMeasurement}
           .includeStatisticsUnitOfMeasurement=${this
             .includeStatisticsUnitOfMeasurement}
+          .includeUnitClass=${this.includeUnitClass}
           .statisticTypes=${this.statisticTypes}
           .statisticIds=${this.statisticIds}
           .label=${this.pickStatisticLabel}

src/data/energy.ts

@@ -600,20 +600,14 @@ export const getEnergySolarForecasts = (hass: HomeAssistant) =>
     type: "energy/solar_forecast",
   });
 
-export const ENERGY_GAS_VOLUME_UNITS = ["m³"];
+const energyGasUnitClass = ["volume", "energy"] as const;
-export const ENERGY_GAS_ENERGY_UNITS = ["kWh"];
+export type EnergyGasUnitClass = typeof energyGasUnitClass[number];
-export const ENERGY_GAS_UNITS = [
-  ...ENERGY_GAS_VOLUME_UNITS,
-  ...ENERGY_GAS_ENERGY_UNITS,
-];
 
-export type EnergyGasUnit = "volume" | "energy";
+export const getEnergyGasUnitClass = (
-
-export const getEnergyGasUnitCategory = (
   prefs: EnergyPreferences,
   statisticsMetaData: Record<string, StatisticsMetaData> = {},
   excludeSource?: string
-): EnergyGasUnit | undefined => {
+): EnergyGasUnitClass | undefined => {
   for (const source of prefs.energy_sources) {
     if (source.type !== "gas") {
       continue;
@@ -622,29 +616,29 @@ export const getEnergyGasUnitCategory = (
       continue;
     }
     const statisticIdWithMeta = statisticsMetaData[source.stat_energy_from];
-    if (statisticIdWithMeta) {
+    if (
-      return ENERGY_GAS_VOLUME_UNITS.includes(
+      energyGasUnitClass.includes(
-        statisticIdWithMeta.statistics_unit_of_measurement
+        statisticIdWithMeta.unit_class as EnergyGasUnitClass
       )
-        ? "volume"
+    ) {
-        : "energy";
+      return statisticIdWithMeta.unit_class as EnergyGasUnitClass;
     }
   }
   return undefined;
 };
 
 export const getEnergyGasUnit = (
+  hass: HomeAssistant,
   prefs: EnergyPreferences,
   statisticsMetaData: Record<string, StatisticsMetaData> = {}
 ): string | undefined => {
-  for (const source of prefs.energy_sources) {
+  const unitClass = getEnergyGasUnitClass(prefs, statisticsMetaData);
-    if (source.type !== "gas") {
+  if (unitClass === undefined) {
-      continue;
+    return undefined;
-    }
-    const statisticIdWithMeta = statisticsMetaData[source.stat_energy_from];
-    if (statisticIdWithMeta?.state_unit_of_measurement) {
-      return statisticIdWithMeta.state_unit_of_measurement;
-    }
   }
-  return undefined;
+  return unitClass === "energy"
+    ? "kWh"
+    : hass.config.unit_system.length === "km"
+    ? "m³"
+    : "ft³";
 };

src/data/recorder.ts

@@ -20,13 +20,13 @@ export interface StatisticValue {
 }
 
 export interface StatisticsMetaData {
-  state_unit_of_measurement: string;
+  statistics_unit_of_measurement: string | null;
-  statistics_unit_of_measurement: string;
   statistic_id: string;
   source: string;
   name?: string | null;
   has_sum: boolean;
   has_mean: boolean;
+  unit_class: string | null;
 }
 
 export type StatisticsValidationResult =
@@ -254,7 +254,7 @@ export const adjustStatisticsSum = (
   statistic_id: string,
   start_time: string,
   adjustment: number,
-  adjustment_unit_of_measurement: string
+  adjustment_unit_of_measurement: string | null
 ): Promise<void> =>
   hass.callWS({
     type: "recorder/adjust_sum_statistics",
@@ -275,3 +275,17 @@ export const getStatisticLabel = (
   }
   return statisticsMetaData?.name || statisticsId;
 };
+
+export const getDisplayUnit = (
+  hass: HomeAssistant,
+  statisticsId: string | undefined,
+  statisticsMetaData: StatisticsMetaData | undefined
+): string | null | undefined => {
+  let unit: string | undefined;
+  if (statisticsId) {
+    unit = hass.states[statisticsId]?.attributes.unit_of_measurement;
+  }
+  return unit === undefined
+    ? statisticsMetaData?.statistics_unit_of_measurement
+    : unit;
+};

src/panels/config/energy/components/ha-energy-gas-settings.ts

@@ -10,7 +10,7 @@ import {
   EnergyPreferencesValidation,
   EnergyValidationIssue,
   GasSourceTypeEnergyPreference,
-  getEnergyGasUnitCategory,
+  getEnergyGasUnitClass,
   saveEnergyPreferences,
 } from "../../../../data/energy";
 import {
@@ -133,7 +133,7 @@ export class EnergyGasSettings extends LitElement {
 
   private _addSource() {
     showEnergySettingsGasDialog(this, {
-      allowedGasUnitCategory: getEnergyGasUnitCategory(
+      allowedGasUnitClass: getEnergyGasUnitClass(
         this.preferences,
         this.statsMetadata
       ),
@@ -152,7 +152,7 @@ export class EnergyGasSettings extends LitElement {
       ev.currentTarget.closest(".row").source;
     showEnergySettingsGasDialog(this, {
       source: { ...origSource },
-      allowedGasUnitCategory: getEnergyGasUnitCategory(
+      allowedGasUnitClass: getEnergyGasUnitClass(
         this.preferences,
         this.statsMetadata,
         origSource.stat_energy_from

src/panels/config/energy/dialogs/dialog-energy-battery-settings.ts

@@ -14,8 +14,7 @@ import { EnergySettingsBatteryDialogParams } from "./show-dialogs-energy";
 import "@material/mwc-button/mwc-button";
 import "../../../../components/entity/ha-statistic-picker";
 
-const energyUnits = ["kWh"];
+const energyUnitClasses = ["energy"];
-const energyDeviceClasses = ["energy"];
 
 @customElement("dialog-energy-battery-settings")
 export class DialogEnergyBatterySettings
@@ -67,8 +66,7 @@ export class DialogEnergyBatterySettings
 
         <ha-statistic-picker
           .hass=${this.hass}
-          .includeStatisticsUnitOfMeasurement=${energyUnits}
+          .includeUnitClass=${energyUnitClasses}
-          .includeDeviceClasses=${energyDeviceClasses}
           .value=${this._source.stat_energy_to}
           .label=${this.hass.localize(
             "ui.panel.config.energy.battery.dialog.energy_into_battery"
@@ -79,8 +77,7 @@ export class DialogEnergyBatterySettings
 
         <ha-statistic-picker
           .hass=${this.hass}
-          .includeStatisticsUnitOfMeasurement=${energyUnits}
+          .includeUnitClass=${energyUnitClasses}
-          .includeDeviceClasses=${energyDeviceClasses}
           .value=${this._source.stat_energy_from}
           .label=${this.hass.localize(
             "ui.panel.config.energy.battery.dialog.energy_out_of_battery"

src/panels/config/energy/dialogs/dialog-energy-device-settings.ts

@@ -14,8 +14,7 @@ import "../../../../components/ha-radio";
 import "../../../../components/ha-formfield";
 import "../../../../components/entity/ha-entity-picker";
 
-const energyUnits = ["kWh"];
+const energyUnitClasses = ["energy"];
-const energyDeviceClasses = ["energy"];
 
 @customElement("dialog-energy-device-settings")
 export class DialogEnergyDeviceSettings
@@ -69,8 +68,7 @@ export class DialogEnergyDeviceSettings
 
         <ha-statistic-picker
           .hass=${this.hass}
-          .includeStatisticsUnitOfMeasurement=${energyUnits}
+          .includeUnitClass=${energyUnitClasses}
-          .includeDeviceClasses=${energyDeviceClasses}
           .label=${this.hass.localize(
             "ui.panel.config.energy.device_consumption.dialog.device_consumption_energy"
           )}

src/panels/config/energy/dialogs/dialog-energy-gas-settings.ts

@@ -5,9 +5,6 @@ import { fireEvent } from "../../../../common/dom/fire_event";
 import "../../../../components/ha-dialog";
 import {
   emptyGasEnergyPreference,
-  ENERGY_GAS_ENERGY_UNITS,
-  ENERGY_GAS_UNITS,
-  ENERGY_GAS_VOLUME_UNITS,
   GasSourceTypeEnergyPreference,
 } from "../../../../data/energy";
 import { HassDialog } from "../../../../dialogs/make-dialog-manager";
@@ -21,7 +18,10 @@ import "../../../../components/ha-radio";
 import "../../../../components/ha-formfield";
 import "../../../../components/ha-textfield";
 import type { HaRadio } from "../../../../components/ha-radio";
-import { getStatisticMetadata } from "../../../../data/recorder";
+import {
+  getStatisticMetadata,
+  getDisplayUnit,
+} from "../../../../data/recorder";
 
 @customElement("dialog-energy-gas-settings")
 export class DialogEnergyGasSettings
@@ -38,7 +38,7 @@ export class DialogEnergyGasSettings
 
   @state() private _pickableUnit?: string;
 
-  @state() private _pickedDisplayUnit?: string;
+  @state() private _pickedDisplayUnit?: string | null;
 
   @state() private _error?: string;
 
@@ -49,7 +49,11 @@ export class DialogEnergyGasSettings
     this._source = params.source
       ? { ...params.source }
       : emptyGasEnergyPreference();
-    this._pickedDisplayUnit = params.metadata?.state_unit_of_measurement;
+    this._pickedDisplayUnit = getDisplayUnit(
+      this.hass,
+      params.source?.stat_energy_from,
+      params.metadata
+    );
     this._costs = this._source.entity_energy_price
       ? "entity"
       : this._source.number_energy_price
@@ -75,9 +79,9 @@ export class DialogEnergyGasSettings
 
     const pickableUnit =
       this._pickableUnit ||
-      (this._params.allowedGasUnitCategory === undefined
+      (this._params.allowedGasUnitClass === undefined
         ? "ft³, m³, Wh, kWh or MWh"
-        : this._params.allowedGasUnitCategory === "energy"
+        : this._params.allowedGasUnitClass === "energy"
         ? "Wh, kWh or MWh"
         : "ft³ or m³");
 
@@ -98,17 +102,12 @@ export class DialogEnergyGasSettings
 
         <ha-statistic-picker
           .hass=${this.hass}
-          .includeStatisticsUnitOfMeasurement=${this._params
+          .includeUnitClass=${this._params.allowedGasUnitClass}
-            .allowedGasUnitCategory === undefined
-            ? ENERGY_GAS_UNITS
-            : this._params.allowedGasUnitCategory === "energy"
-            ? ENERGY_GAS_ENERGY_UNITS
-            : ENERGY_GAS_VOLUME_UNITS}
           .value=${this._source.stat_energy_from}
           .label=${`${this.hass.localize(
             "ui.panel.config.energy.gas.dialog.gas_usage"
           )} (${
-            this._params.allowedGasUnitCategory === undefined
+            this._params.allowedGasUnitClass === undefined
               ? this.hass.localize(
                   "ui.panel.config.energy.gas.dialog.m3_or_kWh"
                 )
@@ -263,14 +262,12 @@ export class DialogEnergyGasSettings
 
   private async _statisticChanged(ev: CustomEvent<{ value: string }>) {
     if (ev.detail.value) {
-      const entity = this.hass.states[ev.detail.value];
+      const metadata = await getStatisticMetadata(this.hass, [ev.detail.value]);
-      if (entity?.attributes.unit_of_measurement) {
+      this._pickedDisplayUnit = getDisplayUnit(
-        this._pickedDisplayUnit = entity.attributes.unit_of_measurement;
+        this.hass,
-      } else {
+        ev.detail.value,
-        this._pickedDisplayUnit = (
+        metadata[0]
-          await getStatisticMetadata(this.hass, [ev.detail.value])
+      );
-        )[0]?.state_unit_of_measurement;
-      }
     } else {
       this._pickedDisplayUnit = undefined;
     }

src/panels/config/energy/dialogs/dialog-energy-grid-flow-settings.ts

@@ -20,8 +20,7 @@ import "../../../../components/ha-formfield";
 import type { HaRadio } from "../../../../components/ha-radio";
 import "../../../../components/entity/ha-entity-picker";
 
-const energyUnits = ["kWh"];
+const energyUnitClasses = ["energy"];
-const energyDeviceClasses = ["energy"];
 
 @customElement("dialog-energy-grid-flow-settings")
 export class DialogEnergyGridFlowSettings
@@ -93,8 +92,7 @@ export class DialogEnergyGridFlowSettings
 
         <ha-statistic-picker
           .hass=${this.hass}
-          .includeStatisticsUnitOfMeasurement=${energyUnits}
+          .includeUnitClass=${energyUnitClasses}
-          .includeDeviceClasses=${energyDeviceClasses}
           .value=${this._source[
             this._params.direction === "from"
               ? "stat_energy_from"

src/panels/config/energy/dialogs/dialog-energy-solar-settings.ts

@@ -22,8 +22,7 @@ import { showConfigFlowDialog } from "../../../../dialogs/config-flow/show-dialo
 import { ConfigEntry, getConfigEntries } from "../../../../data/config_entries";
 import { brandsUrl } from "../../../../util/brands-url";
 
-const energyUnits = ["kWh"];
+const energyUnitClasses = ["energy"];
-const energyDeviceClasses = ["energy"];
 
 @customElement("dialog-energy-solar-settings")
 export class DialogEnergySolarSettings
@@ -79,8 +78,7 @@ export class DialogEnergySolarSettings
 
         <ha-statistic-picker
           .hass=${this.hass}
-          .includeStatisticsUnitOfMeasurement=${energyUnits}
+          .includeUnitClass=${energyUnitClasses}
-          .includeDeviceClasses=${energyDeviceClasses}
           .value=${this._source.stat_energy_from}
           .label=${this.hass.localize(
             "ui.panel.config.energy.solar.dialog.solar_production_energy"

src/panels/config/energy/dialogs/show-dialogs-energy.ts

@@ -2,7 +2,7 @@ import { fireEvent } from "../../../../common/dom/fire_event";
 import {
   BatterySourceTypeEnergyPreference,
   DeviceConsumptionEnergyPreference,
-  EnergyGasUnit,
+  EnergyGasUnitClass,
   EnergyInfo,
   FlowFromGridSourceEnergyPreference,
   FlowToGridSourceEnergyPreference,
@@ -46,7 +46,7 @@ export interface EnergySettingsBatteryDialogParams {
 
 export interface EnergySettingsGasDialogParams {
   source?: GasSourceTypeEnergyPreference;
-  allowedGasUnitCategory?: EnergyGasUnit;
+  allowedGasUnitClass?: EnergyGasUnitClass;
   metadata?: StatisticsMetaData;
   saveCallback: (source: GasSourceTypeEnergyPreference) => Promise<void>;
 }

src/panels/developer-tools/statistics/developer-tools-statistics.ts

@@ -74,12 +74,6 @@ class HaPanelDevStatistics extends SubscribeMixin(LitElement) {
         hidden: this.narrow,
         width: "20%",
       },
-      state_unit_of_measurement: {
-        title: "Display unit",
-        sortable: true,
-        filterable: true,
-        width: "10%",
-      },
       statistics_unit_of_measurement: {
         title: "Statistics unit",
         sortable: true,
@@ -220,12 +214,12 @@ class HaPanelDevStatistics extends SubscribeMixin(LitElement) {
         this._data.push({
           statistic_id: statisticId,
           statistics_unit_of_measurement: "",
-          state_unit_of_measurement: "",
           source: "",
           state: this.hass.states[statisticId],
           issues: issues[statisticId],
           has_mean: false,
           has_sum: false,
+          unit_class: null,
         });
       }
     });

src/panels/developer-tools/statistics/dialog-statistics-adjust-sum.ts

@@ -23,6 +23,7 @@ import "../../../components/ha-svg-icon";
 import {
   adjustStatisticsSum,
   fetchStatistics,
+  getDisplayUnit,
   StatisticValue,
 } from "../../../data/recorder";
 import type { DateTimeSelector, NumberSelector } from "../../../data/selector";
@@ -59,7 +60,7 @@ export class DialogStatisticsFixUnsupportedUnitMetadata extends LitElement {
   };
 
   private _amountSelector = memoizeOne(
-    (unit_of_measurement: string): NumberSelector => ({
+    (unit_of_measurement: string | undefined): NumberSelector => ({
       number: {
         step: 0.01,
         unit_of_measurement,
@@ -135,7 +136,11 @@ export class DialogStatisticsFixUnsupportedUnitMetadata extends LitElement {
     } else {
       const data =
         this._stats5min.length >= 2 ? this._stats5min : this._statsHour;
-      const unit = this._params!.statistic.state_unit_of_measurement;
+      const unit = getDisplayUnit(
+        this.hass,
+        this._params!.statistic.statistic_id,
+        this._params!.statistic
+      );
       const rows: TemplateResult[] = [];
       for (let i = 1; i < data.length; i++) {
         const stat = data[i];
@@ -192,6 +197,11 @@ export class DialogStatisticsFixUnsupportedUnitMetadata extends LitElement {
   }
 
   private _renderAdjustStat() {
+    const unit = getDisplayUnit(
+      this.hass,
+      this._params!.statistic.statistic_id,
+      this._params!.statistic
+    );
     return html`
       <div class="text-content">
         <b>Statistic:</b> ${this._params!.statistic.statistic_id}
@@ -220,9 +230,7 @@ export class DialogStatisticsFixUnsupportedUnitMetadata extends LitElement {
       <ha-selector-number
         label="New Value"
         .hass=${this.hass}
-        .selector=${this._amountSelector(
+        .selector=${this._amountSelector(unit || undefined)}
-          this._params!.statistic.state_unit_of_measurement
-        )}
         .value=${this._amount}
         .disabled=${this._busy}
         @value-changed=${(ev) => {
@@ -299,6 +307,11 @@ export class DialogStatisticsFixUnsupportedUnitMetadata extends LitElement {
   }
 
   private async _fixIssue(): Promise<void> {
+    const unit = getDisplayUnit(
+      this.hass,
+      this._params!.statistic.statistic_id,
+      this._params!.statistic
+    );
     this._busy = true;
     try {
       await adjustStatisticsSum(
@@ -306,7 +319,7 @@ export class DialogStatisticsFixUnsupportedUnitMetadata extends LitElement {
         this._params!.statistic.statistic_id,
         this._chosenStat!.start,
         this._amount! - this._origAmount!,
-        this._params!.statistic.state_unit_of_measurement
+        unit || null
       );
     } catch (err: any) {
       this._busy = false;

src/panels/lovelace/cards/energy/hui-energy-distribution-card.ts

@@ -315,8 +315,11 @@ class HuiEnergyDistrubutionCard
                         ${formatNumber(gasUsage || 0, this.hass.locale, {
                           maximumFractionDigits: 1,
                         })}
-                        ${getEnergyGasUnit(prefs, this._data.statsMetadata) ||
+                        ${getEnergyGasUnit(
-                        "m³"}
+                          this.hass,
+                          prefs,
+                          this._data.statsMetadata
+                        ) || "m³"}
                       </div>
                       <svg width="80" height="30">
                         <path d="M40 0 v30" id="gas" />

src/panels/lovelace/cards/energy/hui-energy-gas-graph-card.ts

@@ -274,7 +274,8 @@ export class HuiEnergyGasGraphCard
       ) as GasSourceTypeEnergyPreference[];
 
     this._unit =
-      getEnergyGasUnit(energyData.prefs, energyData.statsMetadata) || "m³";
+      getEnergyGasUnit(this.hass, energyData.prefs, energyData.statsMetadata) ||
+      "m³";
 
     const datasets: ChartDataset<"bar", ScatterDataPoint[]>[] = [];
 

src/panels/lovelace/cards/energy/hui-energy-sources-table-card.ts

@@ -130,7 +130,8 @@ export class HuiEnergySourcesTableCard
       );
 
     const gasUnit =
-      getEnergyGasUnit(this._data.prefs, this._data.statsMetadata) || "";
+      getEnergyGasUnit(this.hass, this._data.prefs, this._data.statsMetadata) ||
+      "";
 
     const compare = this._data.statsCompare !== undefined;
 

src/panels/lovelace/editor/config-elements/hui-statistics-graph-card-editor.ts

@@ -205,7 +205,7 @@ export class HuiStatisticsGraphCardEditor
       ...this._config,
       stat_types: configured_stat_types,
     };
-    const displayUnit = this._metaDatas?.[0]?.state_unit_of_measurement;
+    const unitClass = this._metaDatas?.[0]?.unit_class;
 
     return html`
       <ha-form
@@ -223,7 +223,7 @@ export class HuiStatisticsGraphCardEditor
           .pickedStatisticLabel=${this.hass!.localize(
             "ui.panel.lovelace.editor.card.statistics-graph.picked_statistic"
           )}
-          .includeDisplayUnitOfMeasurement=${displayUnit}
+          .includeUnitClass=${unitClass}
           .ignoreRestrictionsOnFirstStatistic=${true}
           .value=${this._configEntities}
           .configValue=${"entities"}