From 85cea98e6558296c2194e962c74e03348ba376e6 Mon Sep 17 00:00:00 2001
From: Riley Bode <rbode@bnl.gov>
Date: Wed, 19 Jul 2023 15:21:10 -0400
Subject: [PATCH] Fixed error with create_classes

---
 docs/source/notebooks/elements.ipynb | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/docs/source/notebooks/elements.ipynb b/docs/source/notebooks/elements.ipynb
index bd8320fc..cb5f7d66 100644
--- a/docs/source/notebooks/elements.ipynb
+++ b/docs/source/notebooks/elements.ipynb
@@ -1,6 +1,7 @@
 {
  "cells": [
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -12,6 +13,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -21,6 +23,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -41,7 +44,7 @@
     "connection = SirepoBluesky(\"http://localhost:8000\")\n",
     "\n",
     "data, schema = connection.auth(\"srw\", \"00000002\")\n",
-    "classes, objects = create_classes(connection.data, connection=connection)\n",
+    "classes, objects = create_classes(connection=connection)\n",
     "globals().update(**objects)\n",
     "\n",
     "aperture.horizontalSize.kind = \"hinted\"\n",
@@ -83,6 +86,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -105,7 +109,7 @@
     "connection = SirepoBluesky(\"http://localhost:8000\")\n",
     "\n",
     "data, schema = connection.auth(\"shadow\", \"00000002\")\n",
-    "classes, objects = create_classes(connection.data, connection=connection)\n",
+    "classes, objects = create_classes(connection=connection)\n",
     "globals().update(**objects)\n",
     "\n",
     "print(f\"Number of points before change: {data['models']['simulation']['npoint']}\")\n",
@@ -150,6 +154,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -173,7 +178,7 @@
     "\n",
     "data, schema = connection.auth(\"shadow\", \"00000002\")\n",
     "\n",
-    "classes, objects = create_classes(connection.data, connection=connection)\n",
+    "classes, objects = create_classes(connection=connection)\n",
     "globals().update(**objects)\n",
     "\n",
     "bsr = BeamStatisticsReport(name=\"bsr\", connection=connection)\n",