From ec70dbf6f276a493721f6fd506fd9c56bd65970f Mon Sep 17 00:00:00 2001
From: Benjamin Fineran <bfineran@users.noreply.github.com>
Date: Tue, 13 Feb 2024 09:47:38 -0500
Subject: [PATCH] Update data.py

---
 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