From 4c22a98b0b20c94911c840b97fe54c5938baab37 Mon Sep 17 00:00:00 2001
From: JasperPlant <78851352+JasperPlant@users.noreply.github.com>
Date: Tue, 8 Mar 2022 03:21:13 +0100
Subject: [PATCH] Add entity_category_diagnostics to SGP30 baseline sensors
 (#3272)

---
 esphome/components/sgp30/sensor.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/esphome/components/sgp30/sensor.py b/esphome/components/sgp30/sensor.py
index 2596e0065d..14a078b501 100644
--- a/esphome/components/sgp30/sensor.py
+++ b/esphome/components/sgp30/sensor.py
@@ -13,6 +13,7 @@ from esphome.const import (
     UNIT_PARTS_PER_MILLION,
     UNIT_PARTS_PER_BILLION,
     ICON_MOLECULE_CO2,
+    ENTITY_CATEGORY_DIAGNOSTIC,
 )
 
 DEPENDENCIES = ["i2c"]
@@ -49,10 +50,12 @@ CONFIG_SCHEMA = (
             cv.Optional(CONF_ECO2_BASELINE): sensor.sensor_schema(
                 icon=ICON_MOLECULE_CO2,
                 accuracy_decimals=0,
+                entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
             ),
             cv.Optional(CONF_TVOC_BASELINE): sensor.sensor_schema(
                 icon=ICON_RADIATOR,
                 accuracy_decimals=0,
+                entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
             ),
             cv.Optional(CONF_STORE_BASELINE, default=True): cv.boolean,
             cv.Optional(CONF_BASELINE): cv.Schema(