fix byte array leak (dmlc#1679)

Co-authored-by: Chao Ma <[email protected]>
zeta1999 · Jun 22, 2020 · 90a103e · 90a103e
1 parent 558673e
commit 90a103e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/dgl/runtime/packed_func.h b/include/dgl/runtime/packed_func.h
@@ -779,7 +779,7 @@ class DGLRetValue : public DGLPODValue_ {
   void Clear() {
     if (type_code_ == kNull) return;
     switch (type_code_) {
-      case kStr: delete ptr<std::string>(); break;
+      case kStr: case kBytes: delete ptr<std::string>(); break;
       case kFuncHandle: delete ptr<PackedFunc>(); break;
       case kModuleHandle: delete ptr<Module>(); break;
       case kObjectHandle: delete ptr<std::shared_ptr<Object> >(); break;