From f0c619a4af0810200c6749f63c2474962419a84e Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Thu, 5 Oct 2023 06:05:44 +0100
Subject: [PATCH] Use AsRef<str> for set_one. (#1033)

---
 candle-nn/src/var_map.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/candle-nn/src/var_map.rs b/candle-nn/src/var_map.rs
index d3da84e8..d34cee78 100644
--- a/candle-nn/src/var_map.rs
+++ b/candle-nn/src/var_map.rs
@@ -52,7 +52,7 @@ impl VarMap {
     }
 
     /// Set a named variable to some value.
-    pub fn set_one<K: AsRef<String>, V: AsRef<Tensor>>(&mut self, name: K, value: V) -> Result<()> {
+    pub fn set_one<K: AsRef<str>, V: AsRef<Tensor>>(&mut self, name: K, value: V) -> Result<()> {
         let tensor_data = self.data.lock().unwrap();
         let name = name.as_ref();
         match tensor_data.get(name) {