From 98f4742b14ed19fc127541c4ea8de1e44d79df57 Mon Sep 17 00:00:00 2001 From: Riley Bode Date: Mon, 24 Jul 2023 13:44:46 -0400 Subject: [PATCH] Update sirepo_ophyd.py --- sirepo_bluesky/sirepo_ophyd.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/sirepo_bluesky/sirepo_ophyd.py b/sirepo_bluesky/sirepo_ophyd.py index 38f0d318..1d11df6f 100644 --- a/sirepo_bluesky/sirepo_ophyd.py +++ b/sirepo_bluesky/sirepo_ophyd.py @@ -537,10 +537,8 @@ def create_classes(connection, create_objects=True, extra_model_fields=[]): sirepo_param=i, ) ) - propagation = PropagationConfig(*sirepo_propagation[:]) - classes[inflection.camelize(object_name)] = propagation if create_objects: - objects[object_name] = propagation + objects[object_name] = PropagationConfig(*sirepo_propagation[:]) if sim_type == "srw": post_prop_params = connection.data["models"]["postPropagation"] @@ -555,9 +553,8 @@ def create_classes(connection, create_objects=True, extra_model_fields=[]): sirepo_param=i, ) ) - propagation = PropagationConfig(*sirepo_propagation[:]) - classes[inflection.camelize(object_name)] = propagation + classes["propagation_parameters"] = PropagationConfig if create_objects: - objects[object_name] = propagation + objects[object_name] = PropagationConfig(*sirepo_propagation[:]) return classes, objects