File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -584,15 +584,15 @@ def kpi_scaled(self) -> backend.Tensor:
584584 @functools .cached_property
585585 def media_effects_dist (self ) -> str :
586586 if self .is_national :
587- return constants .NATIONAL_MODEL_SPEC_ARGS [constants .MEDIA_EFFECTS_DIST ]
587+ return constants .NATIONAL_MODEL_SPEC_ARGS [constants .MEDIA_EFFECTS_DIST ] # pytype: disable=bad-return-type
588588 else :
589589 return self ._model_spec .media_effects_dist
590590
591591 @functools .cached_property
592592 def unique_sigma_for_each_geo (self ) -> bool :
593593 if self .is_national :
594594 # Should evaluate to False.
595- return constants .NATIONAL_MODEL_SPEC_ARGS [
595+ return constants .NATIONAL_MODEL_SPEC_ARGS [ # pytype: disable=bad-return-type
596596 constants .UNIQUE_SIGMA_FOR_EACH_GEO
597597 ]
598598 else :
You can’t perform that action at this time.
0 commit comments