empty prediction_strategy cache (#267)

experimental-design · Aug 23, 2023 · 8d258f2 · 8d258f2
1 parent dd27f41
commit 8d258f2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bofire/surrogates/botorch.py b/bofire/surrogates/botorch.py
@@ -55,6 +55,8 @@ def _prepare_for_dump(self):
 
     def _dumps(self) -> str:
         """Dumps the actual model to a string via pickle as this is not directly json serializable."""
+        # empty internal caches to get smaller dumps
+        self.model.prediction_strategy = None
         buffer = io.BytesIO()
         torch.save(self.model, buffer)
         return base64.b64encode(buffer.getvalue()).decode()