From 2cc81a036671371d8d30b0e69b53275168a84e08 Mon Sep 17 00:00:00 2001
From: Anihilatorgunn <n.ushakov@ira-labs.com>
Date: Mon, 22 Jan 2024 19:46:06 +0300
Subject: [PATCH 1/2] Compatibility for custom PatchCombiner

---
 dpipe/predict/shape.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/dpipe/predict/shape.py b/dpipe/predict/shape.py
index 2120ad0..4a4c245 100644
--- a/dpipe/predict/shape.py
+++ b/dpipe/predict/shape.py
@@ -123,9 +123,13 @@ def wrapper(x, *args, **kwargs):
             else:
                 patches = pmap(predict, input_patches, *args, **kwargs)
 
+            prediction_kwargs = {'combiner': combiner, 'get_boxes': get_boxes}
+            if not use_torch:
+                prediction_kwargs['use_torch'] = use_torch
+
             prediction = combine(
                 patches_wrapper(patches), extract(x.shape, input_axis), local_stride, axis,
-                combiner=combiner, get_boxes=get_boxes, use_torch=use_torch
+                **prediction_kwargs
             )
 
             if valid:

From 5580873cee2b0a668355f4764de4b30b27ffb03c Mon Sep 17 00:00:00 2001
From: Anihilatorgunn <n.ushakov@ira-labs.com>
Date: Mon, 22 Jan 2024 19:51:36 +0300
Subject: [PATCH 2/2] version

---
 dpipe/__version__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dpipe/__version__.py b/dpipe/__version__.py
index 73e3bb4..80eb7f9 100644
--- a/dpipe/__version__.py
+++ b/dpipe/__version__.py
@@ -1 +1 @@
-__version__ = '0.3.2'
+__version__ = '0.3.3'