use decorator for converting trials

Author: Moritz-Alexander-Kern

doc/tutorials/unitary_event_analysis.ipynb

@@ -47,6 +47,7 @@
     "import elephant.unitary_event_analysis as ue\n",
     "from elephant.datasets import download_datasets\n",
     "from elephant.trials import TrialsFromBlock, Trials\n",
+    "from elephant.utils import trials_to_list_of_spiketrainlist\n",
     "\n",
     "# Fix random seed to guarantee fixed output\n",
     "random.seed(1224)"
@@ -118,7 +119,7 @@
     "    'suptitle': None,\n",
     "}\n",
     "\n",
-    "\n",
+    "@trials_to_list_of_spiketrainlist\n",
     "def plot_ue(spiketrains, Js_dict, significance_level=0.05,\n",
     "            **plot_params):\n",
     "    \"\"\"\n",
@@ -250,8 +251,6 @@
     "    Refer to `UEA Tutorial <https://elephant.readthedocs.io/en/latest/\n",
     "    tutorials/unitary_event_analysis.html>`_ for real-case scenario.\n",
     "    \"\"\"\n",
-    "    if isinstance(spiketrains, Trials):\n",
-    "      spiketrains = [spiketrains.get_spiketrains_from_trial_as_list(idx) for idx in range(spiketrains.n_trials)]\n",
     "    n_trials = len(spiketrains)\n",
     "    n_neurons = len(spiketrains[0])\n",
     "\n",