fix (#1558)

Co-authored-by: Eugene Khvedchenya <[email protected]> (cherry picked from commit 8a1d255)
Deci-AI · Oct 26, 2023 · 5879c46 · 5879c46
1 parent 28184d3
commit 5879c46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/super_gradients/conversion/conversion_utils.py b/src/super_gradients/conversion/conversion_utils.py
@@ -12,7 +12,7 @@
     (torch.int16, np.int16),
     (torch.int8, np.int8),
     (torch.uint8, np.uint8),
-    (torch.bool, np.bool),
+    (torch.bool, bool),
 ]