From cc4ad61f91e88add538d2f387ac7186a22b72c04 Mon Sep 17 00:00:00 2001
From: Diego Gomez <dgomez@tenstorrent.com>
Date: Tue, 25 Feb 2025 15:01:50 +0000
Subject: [PATCH] Removed the reference to Metalium::Pybinds in tt-train

---
 tt-train/sources/ttml/CMakeLists.txt | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tt-train/sources/ttml/CMakeLists.txt b/tt-train/sources/ttml/CMakeLists.txt
index 5e8bdbac7cc..39cc5f7c034 100644
--- a/tt-train/sources/ttml/CMakeLists.txt
+++ b/tt-train/sources/ttml/CMakeLists.txt
@@ -66,9 +66,9 @@ if(NOT TARGET Metalium::Metal)
         message(FATAL_ERROR "libtt_metal.so not found in ${METALIUM_LIB_PATH}")
     endif()
     if(TTNN_LIBRARY)
-        add_library(Metalium::PYBIND SHARED IMPORTED)
+        add_library(Metalium::TTNN SHARED IMPORTED)
         set_target_properties(
-            Metalium::PYBIND
+            Metalium::TTNN
             PROPERTIES
                 IMPORTED_LOCATION
                     "${TTNN_LIBRARY}"
@@ -101,7 +101,7 @@ target_link_libraries(
         pthread
         atomic
         Metalium::Metal
-        Metalium::PYBIND
+        Metalium::TTNN
         Python::Python
         fmt::fmt-header-only
         magic_enum::magic_enum