Use weights_only for load (#429)

* Use weights_only for load * Update botorch.py * Update random_forest.py
experimental-design · Sep 14, 2024 · 09e9882 · 09e9882
1 parent 0b7a6b1
commit 09e9882
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/bofire/surrogates/botorch.py b/bofire/surrogates/botorch.py
@@ -64,4 +64,4 @@ def _dumps(self) -> str:
  def loads(self, data: str):
  """Loads the actual model from a base64 encoded pickle bytes object and writes it to the `model` attribute."""
  buffer = io.BytesIO(base64.b64decode(data.encode()))
- self.model = torch.load(buffer)
+ self.model = torch.load(buffer, weights_only=False)
diff --git a/bofire/surrogates/random_forest.py b/bofire/surrogates/random_forest.py
@@ -168,4 +168,4 @@ def _dumps(self) -> str:
  def loads(self, data: str):
  """Loads the actual random forest from a base64 encoded pickle bytes object and writes it to the `model` attribute."""
  buffer = io.BytesIO(base64.b64decode(data.encode()))
- self.model = torch.load(buffer)
+ self.model = torch.load(buffer, weights_only=False)