refactor(ml): model sessions (#10559)

2024-06-25 12:00:24 -04:00
parent 6538ad8de7
commit 6356c28f64
11 changed files with 529 additions and 375 deletions
@@ -0,0 +1,5 @@
+from app.schemas import ModelSession
+
+
+def has_batch_axis(session: ModelSession) -> bool:
+    return not isinstance(session.get_inputs()[0].shape[0], int) or session.get_inputs()[0].shape[0] < 0