Fix rust code in docs

oxideai · Apr 16, 2024 · 75dd7af · 75dd7af
1 parent 80f637b
commit 75dd7af
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/device.rs b/src/device.rs
@@ -40,7 +40,8 @@ impl Device {
     ///
     /// Example:
     /// ```rust
-    /// Device::set_default(Device::new(DeviceType::Cpu, 1));
+    /// use mlx::device::{Device, DeviceType};
+    /// Device::set_default(&Device::new(DeviceType::Cpu, 1));
     /// ```
     ///
     /// By default, this is `gpu()`.