Merge branch 'whittington_2020' of https://github.com/SainsburyWellco…

…meCentre/NeuralPlayground into whittington_2020
SainsburyWellcomeCentre · Nov 27, 2024 · a6911d9 · a6911d9
2 parents 3921cf6 + d2cb6c3
commit a6911d9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/examples/agent_examples/whittington_2020_example.ipynb b/examples/agent_examples/whittington_2020_example.ipynb
@@ -24,10 +24,8 @@
     "import importlib\n",
     "import os\n",
     "import pickle\n",
-    "import pandas as pd\n",
-    "import matplotlib.pyplot as plt\n",
-    "import numpy as np\n",
     "\n",
+    "import matplotlib.pyplot as plt\n",
     "import pandas as pd\n",
     "\n",
     "from neuralplayground.comparison import GridScorer\n",

diff --git a/examples/arena_examples/arena_examples.ipynb b/examples/arena_examples/arena_examples.ipynb
@@ -526,8 +526,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from neuralplayground.agents.whittington_2020 import Whittington2020\n",
-    "import neuralplayground.agents.whittington_2020_extras.whittington_2020_parameters as parameters"
+    "import neuralplayground.agents.whittington_2020_extras.whittington_2020_parameters as parameters\n",
+    "from neuralplayground.agents.whittington_2020 import Whittington2020"
    ]
   },
   {

diff --git a/neuralplayground/arenas/discritized_objects.py b/neuralplayground/arenas/discritized_objects.py
@@ -2,7 +2,6 @@
 import matplotlib.pyplot as plt
 import numpy as np
 from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
-from matplotlib.colors import LogNorm
 
 from neuralplayground.arenas.arena_core import Environment
 from neuralplayground.plotting.plot_utils import make_plot_trajectories