diff --git a/dpipe/predict/shape.py b/dpipe/predict/shape.py
index e074941..97186d1 100644
--- a/dpipe/predict/shape.py
+++ b/dpipe/predict/shape.py
@@ -1,7 +1,5 @@
 from functools import wraps
 from typing import Union, Callable, Type
-from more_itertools import batched
-from itertools import chain
 
 import numpy as np