feat(EmissionFactor): fix returned error type

pledge4future · Nov 12, 2024 · 6f4b507 · 6f4b507
1 parent 82ccde2
commit 6f4b507
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/co2calculator/api/emission_factor.py b/co2calculator/api/emission_factor.py
@@ -10,12 +10,11 @@ class EmissionFactor:
     """Stores information on emission factors"""
 
     factor: float
-    source: str
+    source: str = None
 
     def __post_init__(self):
         """Validate the attribute values"""
-        if isinstance(self.factor, (int, float)):
-            assert self.factor >= 0, "Emission factor must be positive"
+        if not isinstance(self.factor, (int, float)):
+            raise TypeError("Emission factor must be a number")
         elif self.factor < 0:
             raise ValueError("Emission factor must be positive")
-        pass