saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-05-25 22:26:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Error for unrecognized options in StanBackend.set_options

Browse source
This commit is contained in:
Ben Letham 2021-03-03 18:04:15 -08:00
parent 4e99ac686e
commit 74b21f7cd9
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
python/fbprophet/models.py
View file

@ -28,8 +28,12 @@ class IStanBackend(ABC):
Specify model options as kwargs.
* newton_fallback [bool]: whether to fallback to Newton if L-BFGS fails
"""
if 'newton_fallback' in kwargs:
self.newton_fallback = kwargs['newton_fallback']
for k, v in kwargs.items():
if k == 'newton_fallback':
self.newton_fallback = v
else:
raise ValueError(f'Unknown option {k}')
@staticmethod
@abstractmethod
@ -291,4 +295,4 @@ class StanBackendEnum(Enum):
try:
return StanBackendEnum[name].value
except KeyError as e:
raise ValueError("Unknown stan backend: {}".format(name)) from e
raise ValueError("Unknown stan backend: {}".format(name)) from e