Type fix

dbt-labs · Sep 5, 2024 · 4af0e56 · 4af0e56
1 parent 3e92abd
commit 4af0e56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metricflow-semantics/metricflow_semantics/specs/time_dimension_spec.py b/metricflow-semantics/metricflow_semantics/specs/time_dimension_spec.py
@@ -155,7 +155,7 @@ def with_grain(self, time_granularity: ExpandedTimeGranularity) -> TimeDimension
         )
 
     def with_grain_and_date_part(  # noqa: D102
-        self, time_granularity: TimeGranularity, date_part: Optional[DatePart]
+        self, time_granularity: ExpandedTimeGranularity, date_part: Optional[DatePart]
     ) -> TimeDimensionSpec:
         return TimeDimensionSpec(
             element_name=self.element_name,