From abf89a560f11621c5d48336823ed179706c49f30 Mon Sep 17 00:00:00 2001
From: Benjamin Fineran <bfineran@users.noreply.github.com>
Date: Tue, 13 Feb 2024 09:48:14 -0500
Subject: [PATCH] [cherry-pick] [Bug] Fix import bug with numpy_log_softmax
 (#1563) #1603

---
 src/deepsparse/utils/data.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/deepsparse/utils/data.py b/src/deepsparse/utils/data.py
index d162c3041f..baaf50b551 100644
--- a/src/deepsparse/utils/data.py
+++ b/src/deepsparse/utils/data.py
@@ -28,6 +28,7 @@
     "split_engine_inputs",
     "join_engine_outputs",
     "prep_for_serialization",
+    "numpy_log_softmax",
 ]
 
 from pydantic import BaseModel