Fix import bug with numpy_log_softmax (#1563)

rahul-tuli · web-flow · commit 3637800618e4 · 2024-01-25T10:58:10.000-05:00
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

Original file line number	Diff line number	Diff line change
`@@ -28,6 +28,7 @@`
`28`	`28`	`"split_engine_inputs",`
`29`	`29`	`"join_engine_outputs",`
`30`	`30`	`"prep_for_serialization",`
	`31`	`+ "numpy_log_softmax",`
`31`	`32`	`]`
`32`	`33`
`33`	`34`	`from pydantic import BaseModel`