From 809bbb5ebb1d92ee53c29ba5664573e1c51a9dd3 Mon Sep 17 00:00:00 2001
From: Yuchen Zhang <134643420+yczhang-nv@users.noreply.github.com>
Date: Thu, 29 Aug 2024 14:09:59 -0700
Subject: [PATCH] revert changes that break the build

---
 .../morpheus/_lib/include/morpheus/messages/control.hpp       | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/python/morpheus/morpheus/_lib/include/morpheus/messages/control.hpp b/python/morpheus/morpheus/_lib/include/morpheus/messages/control.hpp
index 312407d1e9..92456bdf8c 100644
--- a/python/morpheus/morpheus/_lib/include/morpheus/messages/control.hpp
+++ b/python/morpheus/morpheus/_lib/include/morpheus/messages/control.hpp
@@ -162,9 +162,7 @@ enum class MORPHEUS_EXPORT ControlMessageType
 //     std::shared_ptr<TensorMemory> m_tensors;
 // };
 
-namespace morpheus {
-class TensorMemory;
-}
+class MORPHEUS_EXPORT TensorMemory;
 
 // System-clock for better compatibility with pybind11/chrono
 using time_point_t = std::chrono::time_point<std::chrono::system_clock>;