diff --git a/backends/arm/test/modules/test_static_cache.py b/backends/arm/test/modules/test_static_cache.py index c371dedcb69..0950a9a4ebf 100644 --- a/backends/arm/test/modules/test_static_cache.py +++ b/backends/arm/test/modules/test_static_cache.py @@ -80,8 +80,8 @@ def __init__( ) for i in range(len(self.cache.layers)): - self.register_buffer(f"cache_layer_keys_{i}", self.cache.layers[i].keys) - self.register_buffer(f"cache_layer_values_{i}", self.cache.layers[i].values) + self.register_buffer(f"cache_layer_keys_{i}", self.cache.layers[i].keys) # type: ignore[union-attr] + self.register_buffer(f"cache_layer_values_{i}", self.cache.layers[i].values) # type: ignore[union-attr] def forward( self,