Skip to content

Commit 254216b

Browse files
QEDadyThe Meridian Authors
authored andcommitted
Ignore some pytype errors.
PiperOrigin-RevId: 872046423
1 parent 78ce886 commit 254216b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

meridian/model/context.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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:

0 commit comments

Comments
 (0)