Update arg_kv.h

secretflow · Dec 15, 2023 · 5a860d4 · 5a860d4
1 parent 1e35d37
commit 5a860d4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/yacl/utils/spi/argument/arg_kv.h b/yacl/utils/spi/argument/arg_kv.h
@@ -39,14 +39,14 @@ class SpiArg {
   }
 
   template <typename T>
-  &SpiArg operator=(const T &value) {
+  SpiArg& operator=(const T &value) {
     value_ = value;
     return *this;
   }
 
   // Specialized functions of operator=<T>
-  SpiArg operator=(const char *value);
-  SpiArg operator=(const std::string &value);
+  SpiArg& operator=(const char *value);
+  SpiArg& operator=(const std::string &value);
 
   const std::string &Key() const;
   bool HasValue() const;