Skip to content

Commit 296f3d5

Browse files
committed
Update to handle vocs observables as objects
1 parent e7e359f commit 296f3d5

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

optimas/generators/base.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -187,13 +187,9 @@ def _convert_vocs_constraints_to_constraints(
187187
def _convert_vocs_observables_to_parameters(self) -> List[Parameter]:
188188
"""Convert VOCS observables to optimas Parameter objects."""
189189
parameters = []
190-
# Handle both set of strings and dict
191-
for obs_name in self._vocs.observables:
192-
if isinstance(self._vocs.observables, dict):
193-
obs_spec = self._vocs.observables[obs_name]
194-
else:
195-
obs_spec = None
196-
param = Parameter(name=obs_name, dtype=obs_spec)
190+
for obs_name, obs_spec in self._vocs.observables.items():
191+
dtype = obs_spec.dtype
192+
param = Parameter(name=obs_name, dtype=dtype)
197193
parameters.append(param)
198194
return parameters
199195

0 commit comments

Comments
 (0)