Fix import bug with numpy_log_softmax (#1563)

neuralmagic · Feb 13, 2024 · c0891f2 · c0891f2
1 parent bfe7774
commit c0891f2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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