update reference to DerivedFeature-basemodel

diehlbw · Jul 24, 2024 · fcc84d9 · fcc84d9
1 parent 8d1f870
commit fcc84d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/seismometer/configuration/model.py b/src/seismometer/configuration/model.py
@@ -306,13 +306,13 @@ def reduce_cohorts_to_unique_names(cls, cohorts, values) -> list[Cohort]:
         return DataUsage._reduce_derived_features(cohorts, "Cohorts")
 
     @staticmethod
-    def _reduce_derived_features(derived_features, field_name="DataUsage") -> list[Any]:
+    def _reduce_derived_features(derived_features: list[Event | Cohort], field_name="DataUsage") -> list[Any]:
         """
         Reduces the list of derived features to unique names.
 
         Parameters
         ----------
-        derived_features : Optional[list[DerivedFeature]]
+        derived_features : list[Event|Cohort]
             List of derived features.
         field_name : str, optional
             Name of the field for sourcing in logs, by default 'DataUsage'.