imported missing package

beta-vae-normalizing-flows/thoracic-surgery/beta-vae-nice-thoracic-surgery.ipynb

@@ -39,7 +39,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 198,
    "id": "9f0b013a",
    "metadata": {
     "scrolled": true
@@ -59,7 +59,7 @@
        " (0.7775608374378459, 0.8840392521212448, 0.9452007992345052)]"
       ]
      },
-     "execution_count": 2,
+     "execution_count": 198,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -86,6 +86,7 @@
     "from tensorflow.keras.layers import Dense, Flatten, Reshape, Input, BatchNormalization\n",
     "from tensorflow.keras.losses import Loss\n",
     "import scipy\n",
+    "import ks2d2s\n",
     "\n",
     "tfd = tfp.distributions\n",
     "tfb = tfp.bijectors\n",