From 294a8cd05368ca42407106418dab0bb1ea771a0c Mon Sep 17 00:00:00 2001
From: msluszniak <mateusz.sluszniak@swmansion.com>
Date: Thu, 4 Jul 2024 07:01:32 +0000
Subject: [PATCH] Make added functionalities public

---
 lib/ex_vision/utils.ex | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/ex_vision/utils.ex b/lib/ex_vision/utils.ex
index d801618..0cf873e 100644
--- a/lib/ex_vision/utils.ex
+++ b/lib/ex_vision/utils.ex
@@ -156,7 +156,7 @@ defmodule ExVision.Utils do
     process_name |> batched_run([input]) |> hd()
   end
 
-  defp process_bbox(bbox, scales, axes \\ [0]) do
+  def process_bbox(bbox, scales, axes \\ [0]) do
     bbox
     |> Nx.squeeze(axes: axes)
     |> Nx.multiply(scales)
@@ -165,7 +165,7 @@ defmodule ExVision.Utils do
     |> Nx.to_list()
   end
 
-  defp unbatch(batched_value, axes \\ [0]) do
+  def unbatch(batched_value, axes \\ [0]) do
     batched_value |> Nx.squeeze(axes: axes) |> Nx.to_list()
   end
 end