Added tutorial_data to utils

magnipy/__init__.py

@@ -9,3 +9,4 @@
 from .utils.datasets import *
 from .magnipy import Magnipy
 from .diversipy import Diversipy
+from .utils.tutorial_data import *

notebooks/tutorial_data.py → magnipy/utils/tutorial_data.py

@@ -1,19 +1,16 @@
 """File for creating example datasets for user tutorials."""
 
-from importlib import reload
 import numpy as np
 import pandas as pd
 from matplotlib import pyplot as plt
 from sklearn.datasets import make_swiss_roll, make_blobs
 from sklearn.preprocessing import MinMaxScaler
 from magnipy.utils.datasets import (
     sample_points_gaussian,
-    sample_points_gaussian_2,
     sample_points_square,
     hawkes_process,
 )
 from magnipy.utils.plots import plot_points
-from magnipy import Magnipy
 import bisect
 from matplotlib.animation import FuncAnimation
 from magnipy.diversipy import Diversipy

notebooks/mode_dropping.ipynb

@@ -26,18 +26,16 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 7,
    "metadata": {},
    "outputs": [],
    "source": [
     "# Import statements\n",
-    "from matplotlib.animation import FuncAnimation\n",
-    "from magnipy.magnipy import Magnipy\n",
     "from magnipy.diversipy import Diversipy\n",
     "import numpy as np\n",
     "import matplotlib.pyplot as plt\n",
     "import seaborn as sns\n",
-    "from tutorial_data import *"
+    "from magnipy.utils.tutorial_data import *"
    ]
   },
   {
@@ -49,7 +47,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 8,
    "metadata": {},
    "outputs": [
     {
@@ -104,7 +102,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 9,
    "metadata": {},
    "outputs": [
     {
@@ -155,7 +153,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 10,
    "metadata": {},
    "outputs": [
     {
@@ -196,7 +194,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 11,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -213,14 +211,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 12,
    "metadata": {},
    "outputs": [
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "/Users/emily.simons/repos/magnipy/notebooks/tutorial_data.py:473: UserWarning: FigureCanvasAgg is non-interactive, and thus cannot be shown\n",
+      "/home/kat/magnipy/magnipy/magnipy/utils/tutorial_data.py:489: UserWarning: FigureCanvasAgg is non-interactive, and thus cannot be shown\n",
       "  fig.show()\n"
      ]
     },
@@ -285,7 +283,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 13,
    "metadata": {},
    "outputs": [
     {
@@ -337,7 +335,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 14,
    "metadata": {},
    "outputs": [
     {
@@ -372,7 +370,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 15,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -389,7 +387,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 16,
    "metadata": {},
    "outputs": [
     {
@@ -463,17 +461,6 @@
    "source": [
     "Figure from [Metric Space Magnitude for Evaluating the Diversity of Latent Representations](https://arxiv.org/abs/2311.16054). See also for further discussion."
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# To reproduce all animations uncomment the following:\n",
-    "\n",
-    "# create_all_animations()"
-   ]
   }
  ],
  "metadata": {