From 8b87c2bda4e29ce6e8963661c3b1e810686b8ef3 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Mon, 4 Nov 2024 23:15:36 -0500 Subject: [PATCH] fix(MLDataDevices): remove stale import --- lib/MLDataDevices/ext/MLDataDevicesReactantExt.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/MLDataDevices/ext/MLDataDevicesReactantExt.jl b/lib/MLDataDevices/ext/MLDataDevicesReactantExt.jl index a21486bc9..aaf0a4952 100644 --- a/lib/MLDataDevices/ext/MLDataDevicesReactantExt.jl +++ b/lib/MLDataDevices/ext/MLDataDevicesReactantExt.jl @@ -2,7 +2,7 @@ module MLDataDevicesReactantExt using Adapt: Adapt using MLDataDevices: MLDataDevices, Internal, ReactantDevice, CPUDevice, get_device_type -using Reactant: Reactant, XLA, RArray, ConcreteRArray, ConcreteRNumber, TracedRArray, +using Reactant: Reactant, XLA, ConcreteRArray, ConcreteRNumber, TracedRArray, TracedRNumber MLDataDevices.loaded(::Union{ReactantDevice, Type{<:ReactantDevice}}) = true