diff --git a/Project.toml b/Project.toml index 68ad6a8d50..6471e7dfac 100644 --- a/Project.toml +++ b/Project.toml @@ -86,7 +86,7 @@ Preferences = "1.4" PythonCall = "0.9" Random = "1.10" Random123 = "1.7" -ReactantCore = "0.1.9" +ReactantCore = "0.1.9, 0.1.10" Reactant_jll = "0.0.185" ScopedValues = "1.3.0" Scratch = "1.2" diff --git a/lib/ReactantCore/Project.toml b/lib/ReactantCore/Project.toml index bc2feadbac..8b7ff9f894 100644 --- a/lib/ReactantCore/Project.toml +++ b/lib/ReactantCore/Project.toml @@ -1,7 +1,7 @@ name = "ReactantCore" uuid = "a3311ec8-5e00-46d5-b541-4f83e724a433" authors = ["William Moses ", "Valentin Churavy ", "Sergio Sánchez Ramírez ", "Paul Berg ", "Avik Pal "] -version = "0.1.9" +version = "0.1.10" [deps] ExpressionExplorer = "21656369-7473-754a-2065-74616d696c43" diff --git a/lib/ReactantCore/src/ReactantCore.jl b/lib/ReactantCore/src/ReactantCore.jl index 2d47664870..03c541a377 100644 --- a/lib/ReactantCore/src/ReactantCore.jl +++ b/lib/ReactantCore/src/ReactantCore.jl @@ -129,7 +129,7 @@ end ``` """ macro trace(args...) - track_numbers = true + track_numbers = Number expr = first(args) if length(args) > 1 && Meta.isexpr(args[1], :(=)) tn_expr = args[1] @@ -137,11 +137,17 @@ macro trace(args...) error("@trace supports setting track_numbers, but got $(tn_expr)") track_numbers = tn_expr.args[2] + if track_numbers isa Bool + track_numbers = track_numbers ? Number : Union{} + elseif track_numbers == :AbstractFloat + track_numbers = AbstractFloat + else + error("Unknown argument for track_numbers, got $(track_numbers)") + end expr = only(args[2:end]) else expr = only(args) end - track_numbers = track_numbers ? Number : Union{} expr = macroexpand(__module__, expr) if Meta.isexpr(expr, :(=))