add f-string to mosaic memory space error msg

jax-ml · Jan 18, 2025 · 4173842 · 4173842
1 parent 034e967
commit 4173842
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax/_src/pallas/mosaic/lowering.py b/jax/_src/pallas/mosaic/lowering.py
@@ -175,7 +175,7 @@ def _memory_space_to_tpu_memory_space(memory_space: MemorySpace | None
       # Leave the memory space unchanged
       return memory_space
     case _:
-      raise ValueError("Invalid memory space: {memory_space}")
+      raise ValueError(f"Invalid memory space: {memory_space}")
 
 
 def _memory_space_to_mosaic_attribute(memory_space: MemorySpace | None