From 24622b23fef61184b369bcaebdae0eb27881faf2 Mon Sep 17 00:00:00 2001 From: Hana Joo Date: Fri, 21 Aug 2026 05:59:40 -0700 Subject: [PATCH] Add Pyrefly suppressions PiperOrigin-RevId: 968453047 --- .../core/common/keras/compression/algorithms/epr_test.py | 4 ++-- .../python/core/common/keras/compression/internal/optimize.py | 2 +- .../python/core/common/keras/compression/schedules.py | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/tensorflow_model_optimization/python/core/common/keras/compression/algorithms/epr_test.py b/tensorflow_model_optimization/python/core/common/keras/compression/algorithms/epr_test.py index 64316bfcf..8dea88d39 100644 --- a/tensorflow_model_optimization/python/core/common/keras/compression/algorithms/epr_test.py +++ b/tensorflow_model_optimization/python/core/common/keras/compression/algorithms/epr_test.py @@ -182,9 +182,9 @@ def test_reduces_model_size_at_reasonable_accuracy(self): self.assertGreater(accuracy, .9) with self.subTest("compressed_weights_are_smaller"): - original_size = sum( + original_size = sum( # pyrefly: ignore[missing-attribute] map(get_weight_size_in_bytes, model.weights)).numpy() - compressed_size = sum( + compressed_size = sum( # pyrefly: ignore[missing-attribute] map(get_weight_size_in_bytes, compressed_model.weights)).numpy() self.assertLess(compressed_size, 0.01 * original_size) diff --git a/tensorflow_model_optimization/python/core/common/keras/compression/internal/optimize.py b/tensorflow_model_optimization/python/core/common/keras/compression/internal/optimize.py index 7c916f6f8..a207829bb 100644 --- a/tensorflow_model_optimization/python/core/common/keras/compression/internal/optimize.py +++ b/tensorflow_model_optimization/python/core/common/keras/compression/internal/optimize.py @@ -360,7 +360,7 @@ def create_layer_for_training(layer, algorithm): input_shape = layer.input_shape - compressible_weights = algorithm.get_compressible_weights(layer) + compressible_weights = algorithm.get_compressible_weights(layer) # pyrefly: ignore[missing-attribute] # Clone layer for two reasons: # diff --git a/tensorflow_model_optimization/python/core/common/keras/compression/schedules.py b/tensorflow_model_optimization/python/core/common/keras/compression/schedules.py index 6043af6fe..9ae3c9c14 100644 --- a/tensorflow_model_optimization/python/core/common/keras/compression/schedules.py +++ b/tensorflow_model_optimization/python/core/common/keras/compression/schedules.py @@ -87,7 +87,7 @@ def __init__(self, self.begin_step = begin_step self.end_value = end_value self.decay_steps = decay_steps - self.end_step = self.begin_step + self.decay_steps + self.end_step = self.begin_step + self.decay_steps # pyrefly: ignore[unsupported-operation] self.exponent = exponent self.dtype = dtype @@ -111,7 +111,7 @@ def _during_decay(self, step: Union[int, tf.Tensor]) -> tf.Tensor: """Return decayed scheduled value.""" with tf.name_scope(self.name or "PolynomialDecay"): - local_steps = tf.cast(step - self.begin_step, dtype=tf.float32) + local_steps = tf.cast(step - self.begin_step, dtype=tf.float32) # pyrefly: ignore[unsupported-operation] decay_term = tf.math.divide(local_steps, tf.cast(self.decay_steps, dtype=tf.float32)) total_delta = tf.cast(self.start_value - self.end_value, dtype=tf.float32)