Default model update for shadow predictor

deel-ai · Jan 12, 2024 · 04a3b9a · 04a3b9a
1 parent 8716b8d
commit 04a3b9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deel/puncc/api/prediction.py b/deel/puncc/api/prediction.py
@@ -214,7 +214,7 @@ class IdPredictor(BasePredictor):
     :param model: model to be wrapped.
     """
 
-    def __init__(self, model, **kwargs):
+    def __init__(self, model=None, **kwargs):
         self.kwargs = kwargs
         super().__init__(model, is_trained=True)