Update sirepo_ophyd.py

NSLS-II · Jul 24, 2023 · 98f4742 · 98f4742
1 parent fb1fc13
commit 98f4742
Showing 1 changed file with 3 additions and 6 deletions.
diff --git 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