Fixed error with create_classes

NSLS-II · Jul 19, 2023 · 85cea98 · 85cea98
1 parent 2cca9f6
commit 85cea98
Showing 1 changed file with 8 additions and 3 deletions.
diff --git 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",