Fix tests

ecmwf · Aug 6, 2024 · 7611375 · 7611375
1 parent 4aa43f1
commit 7611375
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/covjsonkit/encoder/encoder.py b/covjsonkit/encoder/encoder.py
@@ -1,6 +1,7 @@
 from abc import ABC, abstractmethod
 
 import orjson
+from covjson_pydantic.coverage import CoverageCollection
 from covjson_pydantic.domain import DomainType
 
 from covjsonkit.param_db import get_param_ids, get_params, get_units
@@ -36,9 +37,11 @@ def __init__(self, type, domaintype):
         elif domaintype == "path":
             self.domaintype = "Trajectory"
 
-        # self.pydantic_coverage = CoverageCollection(
-        #    type=type, coverages=[], domainType=self.domaintype, parameters={}, referencing=[]
-        # )
+        # Trajectory not yet implemented in covjson-pydantic
+        if self.domaintype != "Trajectory":
+            self.pydantic_coverage = CoverageCollection(
+            type=type, coverages=[], domainType=self.domaintype, parameters={}, referencing=[]
+            )
         self.parameters = []
 
     def add_parameter(self, param):