chore: apply suggestions from code review

Co-authored-by: Sergio Sánchez Ramírez <[email protected]>
EnzymeAD · Dec 9, 2024 · 101ab7a · 101ab7a
1 parent ca1f1be
commit 101ab7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/TracedRNumber.jl b/src/TracedRNumber.jl
@@ -13,7 +13,7 @@ mutable struct TracedRNumber{T} <: RNumber{T}
 end
 
 get_mlir_data(x::TracedRNumber) = x.mlir_data
-set_mlir_data!(x::TracedRNumber, data) = (x.mlir_data = data)
+set_mlir_data!(x::TracedRNumber, data) = (x.mlir_data = data; return x)
 
 ReactantCore.is_traced(::TracedRNumber) = true