Fix O-1PST not having model name (#173)

airgradienthq · Aug 10, 2024 · edbeab6 · edbeab6
1 parent 9174689
commit edbeab6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/airgradient/util.py b/src/airgradient/util.py
@@ -7,7 +7,7 @@ def get_model_name(model_id: str) -> str | None:
     """Get model name from identifier."""
     if model_id.startswith("I-9PSL"):
         return "AirGradient ONE"
-    if model_id.startswith("O-1PPT"):
+    if model_id.startswith("O-1"):
         return "AirGradient Open Air"
     if "DIY" in model_id:
         return "AirGradient DIY"

diff --git a/tests/test_util.py b/tests/test_util.py
@@ -13,6 +13,7 @@
         ("I-9PSL", "AirGradient ONE"),
         ("I-9PSL-DE", "AirGradient ONE"),
         ("O-1PPT", "AirGradient Open Air"),
+        ("O-1PST", "AirGradient Open Air"),
         ("DIY-PRO-4.3", "AirGradient DIY"),
         ("DIY-PRO-3.7", "AirGradient DIY"),
         ("DIY-BASIC-4.0", "AirGradient DIY"),