Ignore pytype errors produced with --use-functools-partial-overlay

superbobry · tensorflower-gardener · commit 732e7d16235a · 2025-11-05T10:22:03.000-08:00
PiperOrigin-RevId: 828527042
diff --git a/spinoffs/fun_mc/fun_mc/malt.py b/spinoffs/fun_mc/fun_mc/malt.py
@@ -294,7 +294,7 @@ def orig_target_log_prob_fn(x):
       )
     if momentum_sample_fn is None:
       momentum_sample_fn = lambda seed: fun_mc.gaussian_momentum_sample(  # pylint: disable=g-long-lambda
-          state=malt_state.state, seed=seed, named_axis=named_axis
+          state=malt_state.state, seed=seed, named_axis=named_axis  # pytype: disable=attribute-error
       )
     if momentum_refresh_fn is None:
       momentum_refresh_fn = lambda m, seed: _gaussian_momentum_refresh_fn(  # pylint: disable=g-long-lambda

Original file line number	Diff line number	Diff line change
`@@ -294,7 +294,7 @@ def orig_target_log_prob_fn(x):`
`294`	`294`	`)`
`295`	`295`	`if momentum_sample_fn is None:`
`296`	`296`	`momentum_sample_fn = lambda seed: fun_mc.gaussian_momentum_sample( # pylint: disable=g-long-lambda`
`297`		`- state=malt_state.state, seed=seed, named_axis=named_axis`
	`297`	`+ state=malt_state.state, seed=seed, named_axis=named_axis # pytype: disable=attribute-error`
`298`	`298`	`)`
`299`	`299`	`if momentum_refresh_fn is None:`
`300`	`300`	`momentum_refresh_fn = lambda m, seed: _gaussian_momentum_refresh_fn( # pylint: disable=g-long-lambda`