update logic

nv-morpheus · Dec 5, 2024 · febd87c · febd87c
1 parent 18576c3
commit febd87c
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/python/morpheus/morpheus/_lib/src/messages/control.cpp b/python/morpheus/morpheus/_lib/src/messages/control.cpp
@@ -257,11 +257,8 @@ std::shared_ptr<TensorMemory> ControlMessage::tensors()
 
 void ControlMessage::tensors(const std::shared_ptr<TensorMemory>& tensors)
 {
-    m_tensors = tensors;
-    if (tensors)
-    {
-        m_tensor_count = tensors->count;
-    }
+    m_tensors      = tensors;
+    m_tensor_count = tensors ? tensors->count : 0;
 }
 
 TensorIndex ControlMessage::tensor_count()